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:
- ☆12Updated 4 months ago
- TACCL: Guiding Collective Algorithm Synthesis using Communication Sketches☆75Updated 2 years ago
- Medusa: Accelerating Serverless LLM Inference with Materialization [ASPLOS'25]☆29Updated 3 months ago
- ☆23Updated last year
- ☆44Updated 2 months ago
- ☆56Updated 4 years ago
- PerFlow-AI is a programmable performance analysis, modeling, prediction tool for AI system.☆22Updated 4 months ago
- ☆24Updated 3 years ago
- REEF is a GPU-accelerated DNN inference serving system that enables instant kernel preemption and biased concurrent execution in GPU sche…☆100Updated 2 years ago
- Open-source implementation for "Helix: Serving Large Language Models over Heterogeneous GPUs and Network via Max-Flow"☆63Updated 9 months ago
- [ACM EuroSys 2023] Fast and Efficient Model Serving Using Multi-GPUs with Direct-Host-Access☆57Updated 3 weeks ago
- A rust-based benchmark for BlueField SmartNICs.☆29Updated 2 years ago
- ☆50Updated 8 months ago
- ☆25Updated 2 years ago
- ☆15Updated 5 months ago
- Efficient GPU communication over multiple NICs.☆18Updated last month
- ☆15Updated 3 years ago
- Compiler for Dynamic Neural Networks☆46Updated 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
- Repository for MLCommons Chakra schema and tools☆39Updated last year
- ☆75Updated 3 years ago
- Artifact of OSDI '24 paper, ”Llumnix: Dynamic Scheduling for Large Language Model Serving“☆62Updated last year
- AlpaServe: Statistical Multiplexing with Model Parallelism for Deep Learning Serving (OSDI 23)☆84Updated 2 years ago
- A GPU-accelerated DNN inference serving system that supports instant kernel preemption and biased concurrent execution in GPU scheduling.☆43Updated 3 years ago
- [USENIX ATC 2021] Exploring the Design Space of Page Management for Multi-Tiered Memory Systems☆47Updated 3 years ago
- RPCNIC: A High-Performance and Reconfigurable PCIe-attached RPC Accelerator [HPCA2025]☆11Updated 8 months ago
- Bamboo is a system for running large pipeline-parallel DNNs affordably, reliably, and efficiently using spot instances.☆51Updated 2 years ago
- ☆22Updated last year
- Artifacts for our SIGCOMM'22 paper Muri☆42Updated last year
- Deft: A Scalable Tree Index for Disaggregated Memory☆19Updated 4 months ago