nathanbreitsch / torchmocksLinks
Test pytorch code with minimal computational overhead
☆26Updated 2 years ago
Alternatives and similar repositories for torchmocks
Users that are interested in torchmocks are comparing it to the libraries listed below
Sorting:
- Automate issue discovery for your projects against Lightning nightly and releases.☆46Updated 3 months ago
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆66Updated 2 weeks ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 3 weeks ago
- Functional deep learning☆108Updated 2 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆67Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX☆214Updated 2 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated 2 years ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆50Updated 3 weeks ago
- ☆62Updated last year
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆45Updated last year
- Convert scikit-learn models to PyTorch modules☆164Updated last year
- A case study of efficient training of large language models using commodity hardware.☆68Updated 3 years ago
- ☆60Updated 3 years ago
- Image augmentation library for Jax☆39Updated last year
- ☆15Updated 4 years ago
- ☆31Updated 2 months ago
- PyTorch interface for TrueGrad Optimizers☆42Updated 2 years ago
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆68Updated 3 years ago
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"☆13Updated 3 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46Updated 4 years ago
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- ☆47Updated 2 years ago
- MaskedTensors for PyTorch☆38Updated 3 years ago
- Training wheels, side rails, and helicopter parent for your Deep Learning projects in PyTorch☆25Updated last year
- Version control for software 2.0☆65Updated 4 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆55Updated 2 years ago
- Productionize machine learning predictions, with ONNX or without☆66Updated last year