H-Huang / torch_collective_extensionLinks
A minimum demo for PyTorch distributed extension functionality for collectives.
☆15Updated last year
Alternatives and similar repositories for torch_collective_extension
Users that are interested in torch_collective_extension are comparing it to the libraries listed below
Sorting:
- ☆16Updated 8 months ago
- TACCL: Guiding Collective Algorithm Synthesis using Communication Sketches☆79Updated 2 years ago
- Efficient GPU communication over multiple NICs.☆21Updated last month
- Medusa: Accelerating Serverless LLM Inference with Materialization [ASPLOS'25]☆40Updated 7 months ago
- ☆15Updated 3 years ago
- ☆24Updated last year
- Open-source implementation for "Helix: Serving Large Language Models over Heterogeneous GPUs and Network via Max-Flow"☆75Updated 2 months ago
- REEF is a GPU-accelerated DNN inference serving system that enables instant kernel preemption and biased concurrent execution in GPU sche…☆104Updated 3 years ago
- ☆25Updated 3 years ago
- A GPU-accelerated DNN inference serving system that supports instant kernel preemption and biased concurrent execution in GPU scheduling.☆44Updated 3 years ago
- ☆34Updated 2 months ago
- ☆56Updated 4 years ago
- ☆70Updated 2 months ago
- Hi-Speed DNN Training with Espresso: Unleashing the Full Potential of Gradient Compression with Near-Optimal Usage Strategies (EuroSys '2…☆15Updated 2 years ago
- Artifacts for our SIGCOMM'22 paper Muri☆43Updated 2 years ago
- Bamboo is a system for running large pipeline-parallel DNNs affordably, reliably, and efficiently using spot instances.☆55Updated 3 years ago
- ☆20Updated 3 months ago
- PerFlow-AI is a programmable performance analysis, modeling, prediction tool for AI system.☆28Updated 3 weeks ago
- SHADE: Enable Fundamental Cacheability for Distributed Deep Learning Training☆35Updated 2 years ago
- ☆25Updated 2 years ago
- LLM serving cluster simulator☆132Updated last year
- ☆53Updated last year
- GPU-accelerated LLM Training Simulator☆15Updated 6 months ago
- [ACM EuroSys 2023] Fast and Efficient Model Serving Using Multi-GPUs with Direct-Host-Access☆56Updated 5 months ago
- [NSDI25] AutoCCL: Automated Collective Communication Tuning for Accelerating Distributed and Parallel DNN Training☆30Updated 8 months ago
- ☆16Updated last year
- Repository for MLCommons Chakra schema and tools☆39Updated 2 years ago
- ☆36Updated last year
- ☆47Updated last year
- DISB is a new DNN inference serving benchmark with diverse workloads and models, as well as real-world traces.☆57Updated last year