torchcomms: a modern PyTorch communications API
☆344Mar 3, 2026Updated this week
Alternatives and similar repositories for torchcomms
Users that are interested in torchcomms are comparing it to the libraries listed below
Sorting:
- ☆15Feb 24, 2026Updated last week
- Allow torch tensor memory to be released and resumed later☆220Feb 9, 2026Updated 3 weeks ago
- DeepXTrace is a lightweight tool for precisely diagnosing slow ranks in DeepEP-based environments.☆93Jan 16, 2026Updated last month
- Distributed Compiler based on Triton for Parallel Systems☆1,371Feb 13, 2026Updated 2 weeks ago
- Venus Collective Communication Library, supported by SII and Infrawaves.☆138Updated this week
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆52Feb 25, 2026Updated last week
- NCCL Profiling Kit☆152Jul 1, 2024Updated last year
- Tutorial Exercises and Code for GPU Communications Tutorial at HOT Interconnects 2025☆31Oct 22, 2025Updated 4 months ago
- A library to analyze PyTorch traces.☆467Feb 4, 2026Updated last month
- ☆53Feb 24, 2026Updated last week
- NVIDIA Inference Xfer Library (NIXL)☆898Updated this week
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆469Updated this week
- pytorch ucc plugin☆23Jul 8, 2021Updated 4 years ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆475Updated this week
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆478Feb 3, 2026Updated last month
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆177Updated this week
- Estimate MFU for DeepSeekV3☆26Jan 5, 2025Updated last year
- A Micro-benchmarking Tool for HPC Networks☆34Sep 2, 2025Updated 6 months ago
- An experimental communicating attention kernel based on DeepEP.☆35Jul 29, 2025Updated 7 months ago
- Microsoft Collective Communication Library☆385Sep 20, 2023Updated 2 years ago
- Perplexity GPU Kernels☆567Nov 7, 2025Updated 3 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆163Feb 11, 2026Updated 3 weeks ago
- Training hybrid models for dummies.☆29Nov 1, 2025Updated 4 months ago
- ☆32Jul 2, 2025Updated 8 months ago
- Checkpoint-engine is a simple middleware to update model weights in LLM inference engines☆912Updated this week
- ☆52May 19, 2025Updated 9 months ago
- Byted PyTorch Distributed for Hyperscale Training of LLMs and RLs☆938Nov 27, 2025Updated 3 months ago
- Utility scripts for PyTorch (e.g. Make Perfetto show some disappearing kernels, Memory profiler that understands more low-level allocatio…☆90Sep 11, 2025Updated 5 months ago
- Simple PyTorch graph capturing.☆21May 31, 2023Updated 2 years ago
- A Quirky Assortment of CuTe Kernels☆838Updated this week
- A TUI-based utility for real-time monitoring of InfiniBand traffic and performance metrics on the local node☆63Dec 19, 2025Updated 2 months ago
- ATLAHS: An Application-centric Network Simulator Toolchain for AI, HPC, and Distributed Storage☆71Feb 6, 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,224Updated this week
- A hierarchical collective communications library with portable optimizations☆37Dec 8, 2024Updated last year
- ☆83Feb 10, 2026Updated 3 weeks ago
- ByteCheckpoint: An Unified Checkpointing Library for LFMs☆270Feb 2, 2026Updated last month
- Open ABI and FFI for Machine Learning Systems☆351Feb 25, 2026Updated last week
- Efficient Distributed GPU Programming for Exascale, an SC/ISC Tutorial☆353Dec 3, 2025Updated 3 months ago
- This is the repo of "RR-Compound: RDMA-Fused gRPC for Low Latency and High Throughput With an Easy Interface" published in TPDS☆27Mar 12, 2025Updated 11 months ago