ezyang / torchdbg

Related projects ⓘ

Alternatives and complementary repositories for torchdbg

• srush / triton-autodiff
  Experiment of using Tangent to autodiff triton
  ☆71Updated 9 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.
  ☆35Updated 3 months ago
• pytorch-labs / torchfix
  TorchFix - a linter for PyTorch-using code with autofix support
  ☆98Updated last month
• ezyang / tlparse
  TORCH_LOGS parser for PT2
  ☆21Updated 3 weeks ago
• thecharlieblake / lovely-llama
  An implementation of the Llama architecture, to instruct and delight
  ☆21Updated 2 months ago
• graphcore-research / unit-scaling
  A library for unit scaling in PyTorch
  ☆105Updated this week
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆29Updated last week
• UmerHA / triton_util
  Make triton easier
  ☆41Updated 4 months ago
• lianakoleva / no-libtorch-compile
  ☆17Updated 2 weeks ago
• mgmalek / efficient_cross_entropy
  ☆76Updated 5 months ago
• cchan / tccl
  extensible collectives library in triton
  ☆61Updated last month
• lernapparat / torchhacks
  Hacks for PyTorch
  ☆17Updated last year
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• google / jaxonnxruntime
  A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.
  ☆98Updated last month
• softmax1 / Flash-Attention-Softmax-N
  CUDA and Triton implementations of Flash Attention with SoftmaxN.
  ☆66Updated 5 months ago
• cloneofsimo / min-fsdp
  ☆72Updated 4 months ago
• GindaChen / FlexFlashAttention3
  FlexAttention w/ FlashAttention3 Support
  ☆26Updated last month
• drisspg / transformer_nuggets
  A place to store reusable transformer components of my own creation or found on the interwebs
  ☆43Updated this week
• pytorch / torchsnapshot
  A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…
  ☆146Updated this week
• Deep-Learning-Profiling-Tools / triton-viz
  ☆140Updated this week
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆125Updated last week
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆112Updated last year
• neuralmagic / compressed-tensors
  A safetensors extension to efficiently store sparse quantized tensors on disk
  ☆46Updated this week
• stas00 / ml-ways
  ML/DL Math and Method notes
  ☆57Updated 11 months ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆31Updated last year
• facebookresearch / spdl
  Scalable and Performant Data Loading
  ☆42Updated this week
• jxbz / modula
  Scalable neural net training via automatic normalization in the modular norm.
  ☆118Updated 2 months ago
• warner-benjamin / optimi
  Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers
  ☆58Updated 3 months ago
• alexzhang13 / Triton-Puzzles-Solutions
  Personal solutions to the Triton Puzzles
  ☆15Updated 3 months ago