MathInf / toroidal
a lightweight transformer library for PyTorch
β71Updated 3 years ago
Alternatives and similar repositories for toroidal:
Users that are interested in toroidal are comparing it to the libraries listed below
- Yet Another Neural Network Library π€β27Updated last month
- β67Updated last year
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objaxβ128Updated last year
- pytest plugin for a better developer experience when working with the PyTorch test suiteβ44Updated 3 years ago
- Toy implementations of some popular ML optimizers using Python/JAXβ44Updated 3 years ago
- Pytorch Lightning seed project with hydraβ18Updated 4 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transforβ¦β47Updated last year
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-β¦β67Updated last year
- An open source implementation of CLIP.β32Updated 2 years ago
- A collection of code snippets for my PyTorch Lightning projectsβ107Updated 4 years ago
- Example python package with pybind11 cpp extensionβ57Updated 4 years ago
- β25Updated 4 years ago
- β47Updated 4 years ago
- Large dataset storage format for Pytorchβ45Updated 3 years ago
- A GPT, made only of MLPs, in Jaxβ57Updated 3 years ago
- An active learning library for Pytorch based on Lightning-Fabric.β79Updated last year
- β103Updated 4 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'β38Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.β44Updated 4 years ago
- A criticism of a recent paper on buggy image downsampling methods in popular image processing and deep learning libraries.β72Updated 2 years ago
- Simple stochastic weight averaging callback for Kerasβ63Updated 3 years ago
- High performance pytorch modulesβ18Updated 2 years ago
- Implementation of Feedback Transformer in Pytorchβ105Updated 4 years ago
- GAN models implemented with Pytorch Lightning and Hydra configurationβ34Updated 2 years ago
- An Easy To Use PyTorch Computer Vision Libraryβ51Updated last year
- A collection of optimizers, some arcane others well known, for Flax.β29Updated 3 years ago
- Functional deep learningβ108Updated 2 years ago
- PyTorch implementation of L2L execution algorithmβ107Updated 2 years ago
- An implementation of Transformer with Expire-Span, a circuit for learning which memories to retainβ33Updated 4 years ago
- π© Pytorch and Jax code for the Madam optimiser.β51Updated 4 years ago