H-Huang / torch_collective_extension
A minimum demo for PyTorch distributed extension functionality for collectives.
☆11Updated 8 months ago
Alternatives and similar repositories for torch_collective_extension:
Users that are interested in torch_collective_extension are comparing it to the libraries listed below
- ☆12Updated this week
- TACCL: Guiding Collective Algorithm Synthesis using Communication Sketches☆72Updated last year
- Hi-Speed DNN Training with Espresso: Unleashing the Full Potential of Gradient Compression with Near-Optimal Usage Strategies (EuroSys '2…☆15Updated last year
- ☆14Updated 3 years ago
- ☆23Updated 9 months ago
- ☆24Updated last year
- Proteus: A High-Throughput Inference-Serving System with Accuracy Scaling☆10Updated last year
- Bamboo is a system for running large pipeline-parallel DNNs affordably, reliably, and efficiently using spot instances.☆49Updated 2 years ago
- Medusa: Accelerating Serverless LLM Inference with Materialization [ASPLOS'25]☆22Updated 4 months ago
- ☆53Updated 4 years ago
- Managed collective communication service☆18Updated 7 months ago
- Official repository for the paper DynaPipe: Optimizing Multi-task Training through Dynamic Pipelines☆17Updated last year
- PerFlow-AI is a programmable performance analysis, modeling, prediction tool for AI system.☆19Updated this week
- REEF is a GPU-accelerated DNN inference serving system that enables instant kernel preemption and biased concurrent execution in GPU sche…☆94Updated 2 years ago
- Ultra | Ultimate | Unified CCL☆58Updated 2 months ago
- ☆47Updated 3 months ago
- Cupcake: A Compression Scheduler for Scalable Communication-Efficient Distributed Training (MLSys '23)☆9Updated last year
- Arbitrary offloads for RDMA NICs☆89Updated 3 years ago
- ☆23Updated 2 years ago
- RPCNIC: A High-Performance and Reconfigurable PCIe-attached RPC Accelerator [HPCA2025]☆10Updated 4 months ago
- SocksDirect code repository☆19Updated 2 years ago
- Compiler for Dynamic Neural Networks☆46Updated last year
- ☆32Updated 10 months ago
- ThrillerFlow is a Dataflow Analysis and Codegen Framework written in Rust.☆14Updated 5 months ago
- A GPU-accelerated DNN inference serving system that supports instant kernel preemption and biased concurrent execution in GPU scheduling.☆42Updated 2 years ago
- [ACM EuroSys '23] Fast and Efficient Model Serving Using Multi-GPUs with Direct-Host-Access☆56Updated last year
- ☆49Updated 2 years ago
- SHADE: Enable Fundamental Cacheability for Distributed Deep Learning Training☆32Updated 2 years ago
- Repository for MLCommons Chakra schema and tools☆39Updated last year
- [NSDI 2023] TopoOpt: Optimizing the Network Topology for Distributed DNN Training☆28Updated 7 months ago