Alternatives and similar repositories for nccl

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

• lianakoleva / no-libtorch-compile
  ☆21Updated 11 months ago
• imbue-ai / cluster-health
  ☆323Updated last year
• cat-state / tinypar
  ☆20Updated 2 years ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 8 months ago
• fattorib / ZeRO-transformer
  Two implementations of ZeRO-1 optimizer sharding in JAX
  ☆14Updated 2 years ago
• Chillee / lit-llama
  Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code
  ☆10Updated 2 years ago
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆82Updated 2 years ago
• ezyang / torchdbg
  PyTorch centric eager mode debugger
  ☆48Updated last year
• gpu-mode / ring-attention
  ring-attention experiments
  ☆165Updated last year
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 8 months ago
• pytorch / test-infra
  This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …
  ☆105Updated this week
• Jaykef / Triton-nanoGPT
  Custom triton kernels for training Karpathy's nanoGPT.
  ☆19Updated last year
• meta-pytorch / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆226Updated last year
• HazyResearch / train-tk
  train with kittens!
  ☆63Updated last year
• foundation-model-stack / fms-fsdp
  🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…
  ☆280Updated 2 months ago
• nshepperd / flash_attn_jax
  JAX bindings for Flash Attention v2
  ☆103Updated last week
• google-deepmind / asyncdiloco
  ☆47Updated 2 years ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 months ago
• open-lm-engine / accelerated-model-architectures
  A bunch of kernels that might make stuff slower 😉
  ☆75Updated last week
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆133Updated 7 months ago
• erfanzar / jax-flash-attn2
  A flexible and efficient implementation of Flash Attention 2.0 for JAX, supporting multiple backends (GPU/TPU/CPU) and platforms (Triton/…
  ☆34Updated 11 months ago
• mgmalek / efficient_cross_entropy
  ☆124Updated last year
• cchan / tccl
  extensible collectives library in triton
  ☆95Updated 10 months ago
• openxla / tokamax
  Tokamax: A GPU and TPU kernel library.
  ☆170Updated this week
• NVIDIA / nvidia-resiliency-ext
  NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …
  ☆262Updated this week
• graphcore-research / out-of-the-box-fp8-training
  Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.
  ☆46Updated last year
• MatX-inc / seqax
  seqax = sequence modeling + JAX
  ☆170Updated 6 months ago
• meta-pytorch / torchft
  Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
  ☆475Updated last week
• axonn-ai / axonn
  Parallel framework for training and fine-tuning deep neural networks
  ☆70Updated 3 months ago
• DataStates / datastates-llm
  LLM checkpointing for DeepSpeed/Megatron
  ☆24Updated 2 months ago