nathanbreitsch / torchmocks
Test pytorch code with minimal computational overhead
☆25Updated last year
Alternatives and similar repositories for torchmocks:
Users that are interested in torchmocks are comparing it to the libraries listed below
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated 3 weeks ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- ☆15Updated 3 years ago
- ☆30Updated this week
- Automate issue discovery for your projects against Lightning nightly and releases.☆45Updated this week
- Pytorch Lightning seed project with hydra☆18Updated 4 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆26Updated 3 months ago
- AdaCat☆49Updated 2 years ago
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆14Updated 2 months ago
- ☆100Updated 7 months ago
- An easy way to start a python programming environment using GitHub Codespaces.☆15Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Updated last year
- My explorations into editing the knowledge and memories of an attention network☆34Updated 2 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- A metrics library for the JAX ecosystem☆39Updated last year
- Training wheels, side rails, and helicopter parent for your Deep Learning projects in PyTorch☆25Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"☆13Updated 3 years ago
- ☆58Updated 2 years ago
- DEPRECATED--all functionality moved to nbdev☆15Updated 2 years ago
- ☆18Updated last year
- A small library for creating and manipulating custom JAX Pytree classes☆57Updated last year
- Triton Server Component for lightning.ai☆14Updated last year
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated 8 months ago
- MaskedTensors for PyTorch☆38Updated 2 years ago
- In-place debugger for the fastai library and pytorch☆28Updated 3 years ago