tuomaso / train_mnist_fastLinks
How to train a CNN to 99% accuracy on MNIST in less than a second on a laptop
☆64Updated 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:
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- Learning with Signatures☆56Updated 3 years ago
- The code for "No Routing Needed Between Capsules". This repository contains the code used for the experiments detailed in a forthcoming …☆53Updated 4 years ago
- Toy implementations of some popular ML optimizers using Python/JAX☆44Updated 4 years ago
- ☆74Updated 2 years ago
- A small, educational autograd system with deep neural networks support☆13Updated 2 years ago
- ☆16Updated 2 years ago
- Yet Another Neural Network Library 🤔☆27Updated 3 months ago
- A database of over 1.4 billion 3x3 convolution filters extracted from hundreds of diverse CNN models with relevant meta information (CVPR…☆32Updated 2 years ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆252Updated 4 months ago
- Energy-based models in PyTorch☆13Updated 4 years ago
- Code base for SRSGD.☆28Updated 5 years ago
- ToeffiPy is a PyTorch like autograd/deep learning library based only on NumPy.☆16Updated 3 years ago
- ☆36Updated 3 years ago
- [CogSci'21] Study of human inductive biases in CNNs and Transformers.☆43Updated 4 years ago
- Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"☆20Updated last year
- A highly modular PyTorch framework with a focus on Neural Architecture Search (NAS).☆23Updated 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.☆129Updated 7 months ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- Layerwise Batch Entropy Regularization☆23Updated 2 years ago
- ☆47Updated 4 years ago
- The most parameter efficient machine learning models on a few popular benchmarks☆42Updated 3 years ago
- Collection of snippets for PyTorch users☆25Updated 3 years ago
- deep learning from scratch. uses numpy/cupy, trains in GPU, follows pytorch API☆19Updated last year
- Implementing DropPath/StochasticDepth in PyTorch☆16Updated 3 years ago
- ☆50Updated 2 years ago
- FID computation in Jax/Flax.☆27Updated 11 months ago
- CIFAR10 ResNets implemented in JAX+Flax☆12Updated 3 years ago
- Vision Xformers☆21Updated 2 years ago
- Official PyTorch implementation of LilNetX: Lightweight Networks with EXtreme Model Compression and Structured Sparsification☆46Updated 3 years ago