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
- Toy implementations of some popular ML optimizers using Python/JAXβ44Updated 3 years ago
- Yet Another Neural Network Library π€β27Updated last week
- Pytorch Lightning seed project with hydraβ18Updated 4 years ago
- Large dataset storage format for Pytorchβ45Updated 3 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suiteβ44Updated 3 years ago
- β67Updated last year
- β102Updated 4 years ago
- PyTorch implementation of L2L execution algorithmβ107Updated 2 years ago
- Example python package with pybind11 cpp extensionβ57Updated 4 years ago
- β25Updated 4 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objaxβ128Updated last year
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'β38Updated 3 years ago
- A collection of optimizers, some arcane others well known, for Flax.β29Updated 3 years ago
- Deep Probabilistic Programming Course @ DIKUβ57Updated 4 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Moduleβ84Updated last year
- Cyclemoid implementation for PyTorchβ89Updated 3 years ago
- A GPT, made only of MLPs, in Jaxβ57Updated 3 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transforβ¦β47Updated last year
- An open source implementation of CLIP.β32Updated 2 years ago
- π© Pytorch and Jax code for the Madam optimiser.β51Updated 4 years ago
- GAN models implemented with Pytorch Lightning and Hydra configurationβ34Updated 2 years ago
- A collection of code snippets for my PyTorch Lightning projectsβ107Updated 4 years ago
- β47Updated 4 years ago
- π² Iterable dataset resampling in PyTorchβ91Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)β78Updated 4 years ago
- β27Updated 4 years ago
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-β¦β67Updated last year
- AdaCatβ49Updated 2 years ago
- PyTorch Tutorial at the LOD2021 conferenceβ21Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.β44Updated 4 years ago