meta-pytorch / torchfixLinks
TorchFix - a linter for PyTorch-using code with autofix support
☆147Updated 3 weeks ago
Alternatives and similar repositories for torchfix
Users that are interested in torchfix are comparing it to the libraries listed below
Sorting:
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆298Updated this week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆161Updated 2 months ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆239Updated 3 weeks ago
- Torch Distributed Experimental☆117Updated last year
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆59Updated last week
- A library for unit scaling in PyTorch☆130Updated 2 months ago
- Experiment of using Tangent to autodiff triton☆81Updated last year
- PyTorch centric eager mode debugger☆48Updated 9 months ago
- This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic …☆100Updated this week
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- Implementation of Flash Attention in Jax☆217Updated last year
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- A place to store reusable transformer components of my own creation or found on the interwebs☆60Updated 3 weeks ago
- ☆187Updated 2 weeks ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆46Updated last year
- OpTree: Optimized PyTree Utilities☆192Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆123Updated last week
- Fast, Modern, and Low Precision PyTorch Optimizers☆109Updated 2 weeks ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆181Updated 3 weeks ago
- Implementation of a Transformer, but completely in Triton☆274Updated 3 years ago
- Scalable and Performant Data Loading☆302Updated this week
- Load compute kernels from the Hub☆283Updated this week
- Demo of the unit_scaling library, showing how a model can be easily adapted to train in FP8.☆46Updated last year
- Install PyTorch distributions with computation backend auto-detection☆254Updated last month
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆291Updated last year
- Prototype routines for GPU quantization written using PyTorch.☆21Updated last month
- Named Tensors for Legible Deep Learning in JAX☆203Updated last week
- Hacks for PyTorch☆19Updated 2 years ago
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆397Updated this week