tuomaso / train_mnist_fastLinks
How to train a CNN to 99% accuracy on MNIST in less than a second on a laptop
☆69Updated last year
Alternatives and similar repositories for train_mnist_fast
Users that are interested in train_mnist_fast are comparing it to the libraries listed below
Sorting:
- Learning with Signatures☆56Updated 3 years ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆131Updated 10 months ago
- Create animations for the optimization trajectory of neural nets☆159Updated last year
- Yet Another Neural Network Library 🤔☆28Updated last month
- Figures I made during my PhD in Deep Learning, for my models and for context☆83Updated 4 years ago
- Pytorch cyclic cosine decay learning rate scheduler☆49Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- PyTorch implementation of "MLP-Mixer: An all-MLP Architecture for Vision" Tolstikhin et al. (2021)☆31Updated 4 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆85Updated 4 years ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆217Updated 4 years ago
- State-of-the-art data augmentation search algorithms in PyTorch☆47Updated 2 years ago
- Differentiable Algorithms and Algorithmic Supervision.☆116Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 4 years ago
- Implementation of ResMLP, an all MLP solution to image classification, in Pytorch☆200Updated 2 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆66Updated 5 years ago
- Outlining techniques for improving the training performance of your PyTorch model without compromising its accuracy☆129Updated 2 years ago
- ☆134Updated last year
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- code for the ddp tutorial☆32Updated 3 years ago
- Easy-to-use AdaHessian optimizer (PyTorch)☆79Updated 4 years ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆82Updated 2 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆217Updated 2 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆163Updated 2 years ago
- A highly modular PyTorch framework with a focus on Neural Architecture Search (NAS).☆23Updated 3 years ago
- ☆75Updated 2 years ago
- ☆41Updated 4 years ago
- A survey of all the 'X is all you need'☆74Updated 4 years ago
- Automatic learning-rate scheduler☆46Updated 4 years ago