scvi-tools icon indicating copy to clipboard operation
scvi-tools copied to clipboard

Published 20 hours ago • verified publisher icon scverse

Multi-GPU support tracker

Open martinkim0 opened this issue 2 years ago • 0 comments

  • [x] Batch distributed sampler for efficient distributed sparse batching #2102
  • [ ] Jax multi-device with pmap
  • [ ] Multi-device flag in pytest
  • [ ] Integrate with Lightning CI for Azure GPU resources #1314

martinkim0 avatar Jul 10 '23 08:07 martinkim0