Alternatives and similar repositories for pytorch-learning

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

• ray-project / distml
  Distributed ML Optimizer
  ☆32Updated 3 years ago
• spcl / substation
  Research and development for optimizing transformers
  ☆126Updated 4 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆131Updated 3 years ago
• mgmalek / efficient_cross_entropy
  ☆108Updated last year
• stanford-futuredata / stk
  ☆105Updated 9 months ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆96Updated 3 years ago
• asappresearch / flop
  Pytorch library for factorized L0-based pruning.
  ☆45Updated last year
• exists-forall / striped_attention
  ☆38Updated last year
• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆79Updated last year
• clovaai / length-adaptive-transformer
  Official Pytorch Implementation of Length-Adaptive Transformer (ACL 2021)
  ☆101Updated 4 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆54Updated 3 years ago
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆266Updated 3 years ago
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated 10 months ago
• Lightning-AI / forked-pdb
  Python pdb for multiple processes
  ☆44Updated last week
• microsoft / infinibatch
  Efficient, check-pointed data loading for deep learning with massive data sets.
  ☆208Updated last year
• Edward-Sun / gpt-accelera
  Simple and efficient pytorch-native transformer training and inference (batched)
  ☆75Updated last year
• xflash96 / pybind11_package_example
  Example python package with pybind11 cpp extension
  ☆57Updated 4 years ago
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆155Updated 4 years ago
• li-js / gpu_memory_profiling
  Profile the GPU memory usage of every line in a Pytorch code
  ☆82Updated 6 years ago
• JonasGeiping / linear_cross_entropy_loss
  A fusion of a linear layer and a cross entropy loss, written for pytorch in triton.
  ☆67Updated 10 months ago
• ronghanghu / vit_10b_fsdp_example
  See details in https://github.com/pytorch/xla/blob/r1.12/torch_xla/distributed/fsdp/README.md
  ☆24Updated 2 years ago
• huggingface / block_movement_pruning
  Block Sparse movement pruning
  ☆79Updated 4 years ago
• mit-han-lab / neurips-micronet
  [JMLR'20] NeurIPS 2019 MicroNet Challenge Efficient Language Modeling, Champion
  ☆40Updated 4 years ago
• RulinShao / LightSeq
  Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training
  ☆210Updated 9 months ago
• SeanNaren / minGPT
  A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!
  ☆111Updated 2 years ago
• yanring / Megatron-MoE-ModelZoo
  Best practices for testing advanced Mixtral, DeepSeek, and Qwen series MoE models using Megatron Core MoE.
  ☆17Updated this week
• microsoft / varuna
  ☆250Updated 10 months ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated last week
• allenai / sledgehammer
  ☆47Updated 4 years ago
• 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.
  ☆44Updated 10 months ago