meta-pytorch / torchcommsLinks
torchcomms: a modern PyTorch communications API
☆103Updated this week
Alternatives and similar repositories for torchcomms
Users that are interested in torchcomms are comparing it to the libraries listed below
Sorting:
- extensible collectives library in triton☆89Updated 6 months ago
- Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.☆264Updated this week
- ☆92Updated 11 months ago
- NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…☆351Updated last week
- How to ensure correctness and ship LLM generated kernels in PyTorch☆66Updated last week
- TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels☆160Updated last week
- Applied AI experiments and examples for PyTorch☆299Updated 2 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆125Updated 4 months ago
- ☆141Updated 9 months ago
- TPU inference for vLLM, with unified JAX and PyTorch support.☆97Updated last week
- Triton-based Symmetric Memory operators and examples☆48Updated last week
- Collection of kernels written in Triton language☆157Updated 6 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆84Updated last month
- DeeperGEMM: crazy optimized version☆72Updated 5 months ago
- Cataloging released Triton kernels.☆263Updated last month
- ☆240Updated last week
- Perplexity GPU Kernels☆497Updated last month
- A lightweight design for computation-communication overlap.☆181Updated 2 weeks ago
- QuTLASS: CUTLASS-Powered Quantized BLAS for Deep Learning☆119Updated 3 weeks ago
- Fast low-bit matmul kernels in Triton☆381Updated 3 weeks ago
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆55Updated last week
- ☆101Updated 5 months ago
- JAX backend for SGL☆78Updated this week
- ring-attention experiments☆154Updated last year
- ☆65Updated 5 months ago
- A Quirky Assortment of CuTe Kernels☆637Updated 2 weeks ago
- Allow torch tensor memory to be released and resumed later☆150Updated last week
- Github mirror of trition-lang/triton repo.☆86Updated this week
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆138Updated last month
- Efficient Compute-Communication Overlap for Distributed LLM Inference☆59Updated 3 weeks ago