Alternatives and similar repositories for torchdbg

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

• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆80Updated last year
• lianakoleva / no-libtorch-compile
  ☆21Updated 9 months ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 6 months 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.
  ☆46Updated last year
• UmerHA / triton_util
  Make triton easier
  ☆49Updated last year
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆62Updated 2 weeks ago
• meta-pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆161Updated 2 months ago
• cloneofsimo / min-fsdp
  ☆91Updated last year
• meta-pytorch / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆151Updated 3 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆132Updated 4 months ago
• eth-easl / fmengine
  Utilities for Training Very Large Models
  ☆58Updated last year
• pytorch / torchdistx
  Torch Distributed Experimental
  ☆117Updated last year
• meta-pytorch / tlparse
  TORCH_LOGS parser for PT2
  ☆68Updated 3 weeks ago
• facebookresearch / MODel_opt
  Memory Optimizations for Deep Learning (ICML 2023)
  ☆111Updated last year
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆73Updated last year
• mgmalek / efficient_cross_entropy
  ☆121Updated last year
• google-deepmind / asyncdiloco
  ☆47Updated last year
• meta-pytorch / float8_experimental
  This repository contains the experimental PyTorch native float8 training UX
  ☆226Updated last year
• apoorvkh / torchrunx
  Easily run PyTorch on multiple GPUs & machines
  ☆53Updated 2 weeks ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆32Updated 5 months ago
• google / torchax
  torchax is a PyTorch frontend for JAX. It gives JAX the ability to author JAX programs using familiar PyTorch syntax. It also provides JA…
  ☆134Updated last week
• NVIDIA / nsight-python
  Nsight Python is a Python kernel profiling interface based on NVIDIA Nsight Tools
  ☆61Updated last week
• siboehm / ShallowSpeed
  Small scale distributed training of sequential deep learning models, built on Numpy and MPI.
  ☆151Updated 2 years ago
• foundation-model-stack / foundation-model-stack
  🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.
  ☆217Updated last week
• cchan / tccl
  extensible collectives library in triton
  ☆91Updated 8 months ago
• Zyphra / tree_attention
  Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters
  ☆130Updated last year
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆27Updated last year
• cat-state / tinypar
  ☆20Updated 2 years ago
• jax-ml / ml_dtypes
  A stand-alone implementation of several NumPy dtype extensions used in machine learning.
  ☆312Updated last week
• google / aqt
  ☆337Updated last week