A Python library transfers PyTorch tensors between CPU and NVMe
☆125Nov 27, 2024Updated last year
Alternatives and similar repositories for TensorNVMe
Users that are interested in TensorNVMe are comparing it to the libraries listed below
Sorting:
- Performance benchmarking with ColossalAI☆39Jul 6, 2022Updated 3 years ago
- Examples of training models with hybrid parallelism using ColossalAI☆339Mar 23, 2023Updated 2 years ago
- ☆12Apr 30, 2024Updated last year
- ☆30Sep 4, 2023Updated 2 years ago
- ☆28Jul 11, 2021Updated 4 years ago
- A memory efficient DLRM training solution using ColossalAI☆107Nov 22, 2022Updated 3 years ago
- Large-scale model inference.☆627Sep 12, 2023Updated 2 years ago
- A collection of models built with ColossalAI☆32Nov 22, 2022Updated 3 years ago
- Optimizing AlphaFold Training and Inference on GPU Clusters☆613Jul 16, 2024Updated last year
- Scalable PaLM implementation of PyTorch☆190Dec 19, 2022Updated 3 years ago
- Accuracy 77%. Large batch deep learning optimizer LARS for ImageNet with PyTorch and ResNet, using Horovod for distribution. Optional acc…☆38Jun 1, 2021Updated 4 years ago
- ☆19Jan 21, 2026Updated last month
- PyTorch implementation of LAMB for ImageNet/ResNet-50 training☆13May 13, 2021Updated 4 years ago
- Elixir: Train a Large Language Model on a Small GPU Cluster☆15Jun 8, 2023Updated 2 years ago
- ☆12Sep 1, 2023Updated 2 years ago
- A curated list of awesome projects and papers for distributed training or inference☆267Oct 8, 2024Updated last year
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆81Nov 19, 2024Updated last year
- ☆21Jun 6, 2024Updated last year
- ☆21Jul 2, 2022Updated 3 years ago
- ☆14Nov 7, 2025Updated 4 months ago
- ☆216Nov 23, 2025Updated 3 months ago
- TVMScript kernel for deformable attention☆25Dec 15, 2021Updated 4 years ago
- GeminiFS: A Companion File System for GPUs☆72Feb 18, 2025Updated last year
- A baseline repository of Auto-Parallelism in Training Neural Networks☆147Jun 25, 2022Updated 3 years ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆70Mar 20, 2025Updated last year
- ☆12Mar 26, 2024Updated last year
- Efficient AI Inference & Serving☆480Jan 8, 2024Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated last year
- Efficient Auto-scalable Scientific Infrastructure for Engineers and Researchers☆15Sep 8, 2025Updated 6 months ago
- C++17 implementation of einops for libtorch - clear and reliable tensor manipulations with einstein-like notation☆11Oct 16, 2023Updated 2 years ago
- An open-source implementation of Regional Adaptive Sampling (RAS), a novel diffusion model sampling strategy that introduces regional var…☆152Jun 25, 2025Updated 8 months ago
- ☆55Updated this week
- Torch Distributed Experimental☆117Aug 5, 2024Updated last year
- a simple API to use CUPTI☆10Aug 19, 2025Updated 7 months ago
- A prefill & decode disaggregated LLM serving framework with shared GPU memory and fine-grained compute isolation.☆123Dec 25, 2025Updated 2 months ago
- RPCNIC: A High-Performance and Reconfigurable PCIe-attached RPC Accelerator [HPCA2025]☆14Dec 9, 2024Updated last year
- [PACT'24] GraNNDis. A fast and unified distributed graph neural network (GNN) training framework for both full-batch (full-graph) and min…☆10Aug 13, 2024Updated last year
- Fast low-bit matmul kernels in Triton☆438Feb 1, 2026Updated last month
- This is a clone of an SVN repository at http://pagecache-mangagement.googlecode.com/svn/trunk. It had been cloned by http://svn2github.co…☆11May 23, 2013Updated 12 years ago