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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Performance benchmarking with ColossalAI☆39Jul 6, 2022Updated 4 years ago
- A collection of models built with ColossalAI☆33Nov 22, 2022Updated 3 years ago
- Examples of training models with hybrid parallelism using ColossalAI☆339Mar 23, 2023Updated 3 years ago
- ☆12Apr 30, 2024Updated 2 years ago
- ☆30Sep 4, 2023Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆28Jul 11, 2021Updated 5 years ago
- A memory efficient DLRM training solution using ColossalAI☆108Nov 22, 2022Updated 3 years ago
- Large-scale model inference.☆629Sep 12, 2023Updated 2 years ago
- Optimizing AlphaFold Training and Inference on GPU Clusters☆617Jul 16, 2024Updated 2 years ago
- Documentation for Colossal-AI☆25Jun 6, 2025Updated 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 5 years ago
- Elixir: Train a Large Language Model on a Small GPU Cluster☆16Jun 8, 2023Updated 3 years ago
- ☆12Sep 1, 2023Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Code repository for the SOSP'25 paper DCP: Addressing Input Dynamism In Long-Context Training via Dynamic Context Parallelism.☆22Nov 28, 2025Updated 8 months ago
- Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation☆20Jun 11, 2025Updated last year
- A curated list of awesome projects and papers for distributed training or inference☆279Oct 8, 2024Updated last year
- ☆36Feb 18, 2026Updated 5 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆81Nov 19, 2024Updated last year
- ☆21Jun 6, 2024Updated 2 years ago
- ☆15Nov 7, 2025Updated 9 months ago
- ☆21Jul 2, 2022Updated 4 years ago
- TVMScript kernel for deformable attention☆25Dec 15, 2021Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- ☆238Mar 28, 2026Updated 4 months ago
- ☆17Jun 25, 2024Updated 2 years ago
- A baseline repository of Auto-Parallelism in Training Neural Networks☆145Jun 25, 2022Updated 4 years ago
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆72Mar 20, 2025Updated last year
- ☆12Mar 26, 2024Updated 2 years ago
- Efficient AI Inference & Serving☆478Jan 8, 2024Updated 2 years ago
- An open-source implementation of Regional Adaptive Sampling (RAS), a novel diffusion model sampling strategy that introduces regional var…☆155Apr 10, 2026Updated 4 months ago
- ☆25May 29, 2026Updated 2 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- C++17 implementation of einops for libtorch - clear and reliable tensor manipulations with einstein-like notation☆12Oct 16, 2023Updated 2 years ago
- Torch Distributed Experimental☆117Aug 5, 2024Updated 2 years ago
- Official repository of paper [FALQON: Accelerating LoRA Fine-tuning with Low-Bit Floating-Point Arithmetic, NeurIPS 2025]☆21Dec 2, 2025Updated 8 months ago
- GeminiFS: A Companion File System for GPUs☆85Jul 8, 2026Updated last month
- a simple API to use CUPTI☆10Aug 19, 2025Updated 11 months ago
- [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
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆44Nov 4, 2022Updated 3 years ago