torchcomms: a modern PyTorch communications API
☆351Mar 23, 2026Updated this week
Alternatives and similar repositories for torchcomms
Users that are interested in torchcomms are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Tutorial Exercises and Code for GPU Communications Tutorial at HOT Interconnects 2025☆31Oct 22, 2025Updated 5 months ago
- Allow torch tensor memory to be released and resumed later☆225Mar 10, 2026Updated 2 weeks ago
- Venus Collective Communication Library, supported by SII and Infrawaves.☆141Mar 4, 2026Updated 3 weeks ago
- Distributed Compiler based on Triton for Parallel Systems☆1,394Mar 11, 2026Updated last week
- pytorch ucc plugin☆23Jul 8, 2021Updated 4 years ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆490Updated this week
- ☆16Feb 24, 2026Updated last month
- DeepXTrace is a lightweight tool for precisely diagnosing slow ranks in DeepEP-based environments.☆95Jan 16, 2026Updated 2 months ago
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆55Mar 18, 2026Updated last week
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆482Mar 10, 2026Updated 2 weeks ago
- NVIDIA Inference Xfer Library (NIXL)☆945Updated this week
- NCCL Profiling Kit☆152Jul 1, 2024Updated last year
- ☆53Feb 24, 2026Updated last month
- ☆49Aug 27, 2024Updated last year
- Microsoft Collective Communication Library☆387Sep 20, 2023Updated 2 years ago
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 7 months ago
- Estimate MFU for DeepSeekV3☆26Jan 5, 2025Updated last year
- Perplexity GPU Kernels☆564Nov 7, 2025Updated 4 months ago
- ATLAHS: An Application-centric Network Simulator Toolchain for AI, HPC, and Distributed Storage☆73Mar 10, 2026Updated 2 weeks ago
- A library to analyze PyTorch traces.☆474Mar 17, 2026Updated last week
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆182Updated this week
- Training hybrid models for dummies.☆29Nov 1, 2025Updated 4 months ago
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆487Mar 17, 2026Updated last week
- TACCL: Guiding Collective Algorithm Synthesis using Communication Sketches☆80Jul 25, 2023Updated 2 years ago
- A Quirky Assortment of CuTe Kernels☆863Updated this week
- Simple PyTorch graph capturing.☆21May 31, 2023Updated 2 years ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆170Feb 11, 2026Updated last month
- Byted PyTorch Distributed for Hyperscale Training of LLMs and RLs☆1,000Mar 3, 2026Updated 3 weeks ago
- A Micro-benchmarking Tool for HPC Networks☆34Sep 2, 2025Updated 6 months ago
- Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…☆94Sep 11, 2025Updated 6 months ago
- Slowdown prediction module of Echo: Simulating Distributed Training at Scale☆13May 17, 2025Updated 10 months ago
- Optimized primitives for collective multi-GPU communication☆4,531Mar 17, 2026Updated last week
- Checkpoint-engine is a simple middleware to update model weights in LLM inference engines☆925Feb 28, 2026Updated 3 weeks ago
- UCCL is an efficient communication library for GPUs, covering collectives, P2P (e.g., KV cache transfer, RL weight transfer), and EP (e.g…☆1,240Updated this week
- ☆32Jul 2, 2025Updated 8 months ago
- ☆52May 19, 2025Updated 10 months ago
- ☆139Aug 18, 2025Updated 7 months ago
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆271Feb 2, 2026Updated last month
- Asynchronous Rust bindings for UCX☆79Apr 29, 2025Updated 10 months ago