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
- ☆15Updated 3 years ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated last month
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated last year
- ☆18Updated last year
- AdaCat☆49Updated 2 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- Configuration classes enabling Hydra to configure and manage Pytorch Lightning projects.☆41Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 4 months ago
- PyTorch interface for TrueGrad Optimizers☆42Updated last year
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 3 years ago
- Automate issue discovery for your projects against Lightning nightly and releases.☆46Updated this week
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- Pytorch Lightning seed project with hydra☆18Updated 4 years ago
- ☆31Updated last month
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- A case study of efficient training of large language models using commodity hardware.☆68Updated 2 years ago
- An easy way to start a python programming environment using GitHub Codespaces.☆15Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- High performance pytorch modules☆18Updated 2 years ago
- ☆25Updated 3 years ago
- Image augmentation library for Jax☆37Updated 10 months ago
- ☆101Updated 8 months ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- A metrics library for the JAX ecosystem☆39Updated last year
- Companion code for a tutorial on using Hydra.☆29Updated 3 years ago
- Utilities for PyTorch distributed☆23Updated this week
- Yet Another Neural Network Library 🤔☆25Updated last month
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆60Updated last month