pytorch-labs / torchfix
TorchFix - a linter for PyTorch-using code with autofix support
☆103Updated this week
Related projects ⓘ
Alternatives and complementary repositories for torchfix
- Multidimensional indexing for tensors☆113Updated last year
- Lightning HPO & Training Studio App☆18Updated last year
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆51Updated this week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆146Updated 2 weeks ago
- Torch Distributed Experimental☆116Updated 3 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…☆216Updated last week
- Named Tensors for Legible Deep Learning in JAX☆153Updated this week
- Scalable neural net training via automatic normalization in the modular norm.☆121Updated 3 months ago
- A library for unit scaling in PyTorch☆105Updated 2 weeks ago
- PyTorch centric eager mode debugger☆43Updated 4 months ago
- Experiment of using Tangent to autodiff triton☆72Updated 9 months ago
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆215Updated this week
- ☆73Updated 4 months ago
- Functional deep learning☆106Updated last year
- A place to store reusable transformer components of my own creation or found on the interwebs☆44Updated 2 weeks ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆328Updated last month
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆58Updated 4 months ago
- Implementation of Flash Attention in Jax☆196Updated 8 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆98Updated 2 months ago
- This is a port of Mistral-7B model in JAX☆30Updated 4 months ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆42Updated 5 months 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…☆175Updated last week
- A Python package of computer vision models for the Equinox ecosystem.☆102Updated 4 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆343Updated 3 weeks ago
- ☆128Updated this week
- ☆105Updated 2 weeks ago
- Efficient CUDA kernels for training convolutional neural networks with PyTorch.☆35Updated 3 weeks ago
- A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …☆176Updated 6 months ago