Alternatives and similar repositories for varuna

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

• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆161Updated 2 months ago
• spcl / substation
  Research and development for optimizing transformers
  ☆131Updated 4 years ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆276Updated 3 years ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆216Updated last week
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆428Updated 2 weeks ago
• stanford-futuredata / stk
  ☆113Updated last year
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• anyscale / llm-continuous-batching-benchmarks
  ☆122Updated last year
• awslabs / slapo
  A schedule language for large model training
  ☆151Updated 2 months ago
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆217Updated last year
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆135Updated 5 months ago
• meta-pytorch / torchx
  TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…
  ☆400Updated this week
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆275Updated 2 weeks ago
• meta-pytorch / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆223Updated last year
• hpcaitech / TensorNVMe
  A Python library transfers PyTorch tensors between CPU and NVMe
  ☆121Updated 11 months ago
• meta-pytorch / applied-ai
  Applied AI experiments and examples for PyTorch
  ☆305Updated 2 months ago
• DS3Lab / DT-FM
  ☆93Updated 3 years ago
• shawntan / scattermoe
  Triton-based implementation of Sparse Mixture of Experts.
  ☆248Updated last month
• cchan / tccl
  extensible collectives library in triton
  ☆91Updated 7 months ago
• pytorch / PiPPy
  Pipeline Parallelism for PyTorch
  ☆784Updated last year
• zhuohan123 / terapipe
  ☆77Updated 4 years ago
• google / aqt
  ☆337Updated 2 weeks ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆137Updated 3 years ago
• Deep-Learning-Profiling-Tools / triton-viz
  ☆247Updated this week
• yandex-research / DeDLOC
  Official code for "Distributed Deep Learning in Open Collaborations" (NeurIPS 2021)
  ☆118Updated 3 years ago
• albanD / subclass_zoo
  ☆182Updated last year
• saareliad / FTPipe
  FTPipe and related pipeline model parallelism research.
  ☆43Updated 2 years 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 …
  ☆234Updated this week
• mgmalek / efficient_cross_entropy
  ☆121Updated 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…
  ☆272Updated 2 weeks ago