Alternatives and similar repositories for torchdynamo

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

• ELS-RD / kernl
  Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…
  ☆1,571Updated last year
• hidet-org / hidet
  An open-source efficient deep learning framework/compiler, written in python.
  ☆703Updated this week
• pytorch / PiPPy
  Pipeline Parallelism for PyTorch
  ☆768Updated 10 months ago
• pytorch / kineto
  A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
  ☆821Updated this week
• pytorch / torchx
  TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…
  ☆367Updated this week
• pytorch / functorch
  functorch is JAX-like composable function transforms for PyTorch.
  ☆1,432Updated this week
• facebookresearch / HolisticTraceAnalysis
  A library to analyze PyTorch traces.
  ☆387Updated last week
• dmlc / dlpack
  common in-memory tensor structure
  ☆1,014Updated last week
• pytorch / data
  A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
  ☆1,206Updated this week
• jiazhihao / TASO
  The Tensor Algebra SuperOptimizer for Deep Learning
  ☆715Updated 2 years ago
• facebookresearch / fairscale
  PyTorch extensions for high performance and large scale training.
  ☆3,331Updated last month
• pytorch / torcharrow
  High performance model preprocessing library on PyTorch
  ☆650Updated last year
• NVIDIA / TransformerEngine
  A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper, Ada and Bla…
  ☆2,491Updated this week
• pytorch / FBGEMM
  FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
  ☆1,383Updated this week
• lucidrains / triton-transformer
  Implementation of a Transformer, but completely in Triton
  ☆268Updated 3 years ago
• facebookresearch / bitsandbytes
  Library for 8-bit optimizers and quantization routines.
  ☆716Updated 2 years ago
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆930Updated this week
• jax-ml / jax-triton
  jax-triton contains integrations between JAX and OpenAI Triton
  ☆400Updated 2 weeks ago
• pytorch / benchmark
  TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
  ☆948Updated 2 weeks ago
• thuml / depyf
  depyf is a tool to help you understand and adapt to PyTorch compiler torch.compile.
  ☆689Updated 2 months ago
• NVIDIA / PyProf
  A GPU performance profiling tool for PyTorch models
  ☆503Updated 3 years ago
• microsoft / Tutel
  Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4
  ☆842Updated this week
• microsoft / nnfusion
  A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
  ☆989Updated 9 months ago
• llvm / torch-mlir
  The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.
  ☆1,562Updated last week
• kakaobrain / torchgpipe
  A GPipe implementation in PyTorch
  ☆843Updated 10 months ago
• albanD / subclass_zoo
  ☆168Updated last year
• BobMcDear / attorch
  A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.
  ☆554Updated this week
• pytorch / xla
  Enabling PyTorch on XLA Devices (e.g. Google TPU)
  ☆2,623Updated this week
• Azure / MS-AMP
  Microsoft Automatic Mixed Precision Library
  ☆609Updated 8 months ago
• google / aqt
  ☆317Updated this week