Alternatives and similar repositories for nccl

Users that are interested in nccl are comparing it to the libraries listed below

• imbue-ai / cluster-health
  ☆314Updated 11 months ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 5 months ago
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆200Updated this week
• coreweave / ml-containers
  ☆38Updated this week
• pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆377Updated this week
• zinccat / flaxattention
  ☆20Updated 9 months ago
• gau-nernst / learn-cuda
  Learn CUDA with PyTorch
  ☆33Updated 3 weeks ago
• coreweave / tensorizer
  Module, Model, and Tensor Serialization/Deserialization
  ☆250Updated this week
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆258Updated 2 weeks ago
• gpu-mode / ring-attention
  ring-attention experiments
  ☆147Updated 9 months ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 2 months ago
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated 2 years ago
• NVIDIA / cuda-checkpoint
  CUDA checkpoint and restore utility
  ☆357Updated 6 months ago
• huggingface / kernels
  Load compute kernels from the Hub
  ☆233Updated this week
• pytorch-labs / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆224Updated last year
• lucidrains / flash-attention-jax
  Implementation of Flash Attention in Jax
  ☆215Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• AI-Hypercomputer / xpk
  xpk (Accelerated Processing Kit, pronounced x-p-k,) is a software tool to help Cloud developers to orchestrate training jobs on accelerat…
  ☆136Updated this week
• cloneofsimo / min-fsdp
  ☆83Updated last year
• Jaykef / Triton-nanoGPT
  Custom triton kernels for training Karpathy's nanoGPT.
  ☆19Updated 9 months ago
• cat-state / tinypar
  ☆20Updated 2 years ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 2 months ago
• zinccat / Awesome-Triton-Kernels
  Collection of kernels written in Triton language
  ☆142Updated 4 months ago
• pytorch-labs / monarch
  PyTorch Single Controller
  ☆345Updated this week
• open-lm-engine / flash-model-architectures
  A bunch of kernels that might make stuff slower 😉
  ☆56Updated last week
• Chillee / lit-llama
  Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code
  ☆11Updated last year
• EleutherAI / nanoGPT-mup
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆149Updated last month
• cchan / tccl
  extensible collectives library in triton
  ☆88Updated 4 months ago
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆63Updated 4 months ago
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆158Updated last month