Alternatives and similar repositories for pytorch-learning

Users that are interested in pytorch-learning are comparing it to the libraries listed below

• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆211Updated 2 years ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆279Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆164Updated 3 weeks ago
• ray-project / distml
  Distributed ML Optimizer
  ☆35Updated 4 years ago
• microsoft / varuna
  ☆252Updated last year
• mgmalek / efficient_cross_entropy
  ☆124Updated last year
• li-js / gpu_memory_profiling
  Profile the GPU memory usage of every line in a Pytorch code
  ☆83Updated 7 years ago
• spcl / substation
  Research and development for optimizing transformers
  ☆131Updated 4 years ago
• pytorch / rfcs
  PyTorch RFCs (experimental)
  ☆138Updated 8 months ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆97Updated 4 years ago
• epfml / dynamic-sparse-flash-attention
  ☆150Updated 2 years ago
• jaymody / speculative-sampling
  Simple implementation of Speculative Sampling in NumPy for GPT-2.
  ☆99Updated 2 years ago
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆79Updated last year
• microsoft / mutransformers
  some common Huggingface transformers in maximal update parametrization (µP)
  ☆87Updated 3 years ago
• google / praxis
  ☆192Updated last week
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆137Updated 3 years ago
• stanford-futuredata / stk
  ☆115Updated last year
• tanyuqian / redco
  NAACL '24 (Best Demo Paper RunnerUp) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Training and Inference
  ☆69Updated last year
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆219Updated last week
• hpcaitech / PaLM-colossalai
  Scalable PaLM implementation of PyTorch
  ☆190Updated 3 years ago
• JonasGeiping / linear_cross_entropy_loss
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆75Updated last year
• SeanNaren / minGPT
  A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!
  ☆112Updated 2 years ago
• Sea-Snell / JAXSeq
  Train very large language models in Jax.
  ☆210Updated 2 years ago
• aniquetahir / JORA
  JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)
  ☆36Updated last year
• anyscale / llm-continuous-batching-benchmarks
  ☆125Updated last year
• Lightning-AI / forked-pdb
  Python pdb for multiple processes
  ☆80Updated 8 months ago
• kshitij12345 / torchnnprofiler
  Context Manager to profile the forward and backward times of PyTorch's nn.Module
  ☆83Updated 2 years ago
• microsoft / tensorflow-rematerialization
  Implementation of a Tensorflow XLA rematerialization pass
  ☆15Updated 6 years ago
• asappresearch / flop
  Pytorch library for factorized L0-based pruning.
  ☆45Updated 2 years ago