meta-pytorch / torchcommsLinks
torchcomms: a modern PyTorch communications API
☆327Updated this week
Alternatives and similar repositories for torchcomms
Users that are interested in torchcomms are comparing it to the libraries listed below
Sorting:
- Perplexity open source garden for inference technology☆359Updated last month
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆462Updated last month
- Perplexity GPU Kernels☆554Updated 3 months ago
- Genai-bench is a powerful benchmark tool designed for comprehensive token-level performance evaluation of large language model (LLM) serv…☆263Updated this week
- ☆159Updated last year
- An early research stage expert-parallel load balancer for MoE models based on linear programming.☆495Updated 2 months ago
- Allow torch tensor memory to be released and resumed later☆216Updated 3 weeks ago
- ☆342Updated last week
- Accelerating MoE with IO and Tile-aware Optimizations☆569Updated 3 weeks ago
- A lightweight design for computation-communication overlap.☆219Updated 2 weeks ago
- Helpful kernel tutorials and examples for tile-based GPU programming☆630Updated this week
- Microsoft Collective Communication Library☆66Updated last year
- Dynamic Memory Management for Serving LLMs without PagedAttention☆457Updated 8 months ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆457Updated last week
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆324Updated this week
- ☆104Updated last year
- Autonomous GPU Kernel Generation via Deep Agents☆228Updated this week
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆82Updated last year
- Open ABI and FFI for Machine Learning Systems☆333Updated this week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆194Updated this week
- A NCCL extension library, designed to efficiently offload GPU memory allocated by the NCCL communication library.☆90Updated last month
- High-performance distributed data shuffling (all-to-all) library for MoE training and inference☆111Updated last month
- DeepXTrace is a lightweight tool for precisely diagnosing slow ranks in DeepEP-based environments.☆92Updated 3 weeks ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆161Updated 4 months ago
- Github mirror of trition-lang/triton repo.☆128Updated this week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆262Updated this week
- ☆84Updated 3 months ago
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆168Updated this week
- ☆286Updated this week
- KV cache store for distributed LLM inference☆390Updated 2 months ago