Alternatives and similar repositories for distributed-training-and-deepspeed

Users that are interested in distributed-training-and-deepspeed are comparing it to the libraries listed below

• lessw2020 / transformer_central
  Various transformers for FSDP research
  ☆38Updated 3 years ago
• AnswerDotAI / cold-compress
  Cold Compress is a hackable, lightweight, and open-source toolkit for creating and benchmarking cache compression methods built on top of…
  ☆146Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆218Updated this week
• mgmalek / efficient_cross_entropy
  ☆124Updated last year
• jaymody / speculative-sampling
  Simple implementation of Speculative Sampling in NumPy for GPT-2.
  ☆99Updated 2 years ago
• llm-efficiency-challenge / neurips_llm_efficiency_challenge
  NeurIPS Large Language Model Efficiency Challenge: 1 LLM + 1GPU + 1Day
  ☆260Updated 2 years ago
• hamelsmu / llama-inference
  experiments with inference on llama
  ☆103Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆81Updated 2 years ago
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆70Updated 2 weeks ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• fw-ai / benchmark
  Benchmark suite for LLMs from Fireworks.ai
  ☆86Updated 2 weeks ago
• anyscale / llm-continuous-batching-benchmarks
  ☆124Updated last year
• xrsrke / pipegoose
  Large scale 4D parallelism pre-training for 🤗 transformers in Mixture of Experts *(still work in progress)*
  ☆86Updated 2 years ago
• Jokeren / triton-samples
  ☆28Updated last year
• MekkCyber / TritonAcademy
  A repository to unravel the language of GPUs, making their kernel conversations easy to understand
  ☆195Updated 7 months ago
• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆164Updated 2 weeks ago
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆154Updated 2 years ago
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆73Updated last year
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆278Updated 3 years ago
• 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
• cloneofsimo / min-fsdp
  ☆92Updated last year
• gpu-mode / profiling-cuda-in-torch
  ☆178Updated last year
• tspeterkim / paged-attention-minimal
  a minimal cache manager for PagedAttention, on top of llama3.
  ☆130Updated last year
• rasbt / pytorch-memory-optim
  This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…
  ☆92Updated 2 years ago
• sgugger / torchdynamo-tests
  ☆20Updated 3 years ago
• SeunghyunSEO / optimized_hf_llama_class_for_training
  ☆48Updated last year
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated 2 years ago
• lianakoleva / no-libtorch-compile
  ☆21Updated 10 months ago
• UmerHA / triton_util
  Make triton easier
  ☆50Updated last year
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆66Updated 2 years ago