H-Huang / torch_collective_extensionLinks
A minimum demo for PyTorch distributed extension functionality for collectives.
☆14Updated 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:
- ☆15Updated 7 months ago
- TACCL: Guiding Collective Algorithm Synthesis using Communication Sketches☆78Updated 2 years ago
- ☆23Updated last year
- Open-source implementation for "Helix: Serving Large Language Models over Heterogeneous GPUs and Network via Max-Flow"☆74Updated 2 months ago
- Efficient GPU communication over multiple NICs.☆21Updated last month
- REEF is a GPU-accelerated DNN inference serving system that enables instant kernel preemption and biased concurrent execution in GPU sche…☆102Updated 2 years ago
- Medusa: Accelerating Serverless LLM Inference with Materialization [ASPLOS'25]☆40Updated 7 months ago
- ☆15Updated 3 years ago
- ☆46Updated last year
- ☆44Updated 4 years ago
- Managed collective communication service☆23Updated last year
- ☆65Updated last month
- Compiler for Dynamic Neural Networks☆46Updated 2 years ago
- ☆20Updated 2 months ago
- ☆33Updated 2 months ago
- Bamboo is a system for running large pipeline-parallel DNNs affordably, reliably, and efficiently using spot instances.☆54Updated 3 years ago
- ☆25Updated 2 years 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
- Repository for MLCommons Chakra schema and tools☆39Updated last year
- ☆16Updated last year
- Artifact of OSDI '24 paper, ”Llumnix: Dynamic Scheduling for Large Language Model Serving“☆64Updated last year
- ☆53Updated 11 months ago
- ☆25Updated 3 years ago
- Artifact for "Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Serving" [SOSP '24]☆25Updated last year
- ☆36Updated last year
- PerFlow-AI is a programmable performance analysis, modeling, prediction tool for AI system.☆25Updated last week
- GPU-accelerated LLM Training Simulator☆47Updated 5 months ago
- A GPU-accelerated DNN inference serving system that supports instant kernel preemption and biased concurrent execution in GPU scheduling.☆44Updated 3 years ago
- Paella: Low-latency Model Serving with Virtualized GPU Scheduling☆66Updated last year
- Supplemental materials for The ASPLOS 2025 / EuroSys 2025 Contest on Intra-Operator Parallelism for Distributed Deep Learning☆24Updated 7 months ago