a-nico / ArrayFlow
☆15Updated 4 years ago
Alternatives and similar repositories for ArrayFlow:
Users that are interested in ArrayFlow are comparing it to the libraries listed below
- Simple implementation of the LSUV initialization in PyTorch☆58Updated last year
- ☆57Updated 2 years ago
- EfficientNet, MobileNetV3, MobileNetV2, MixNet, etc in JAX w/ Flax Linen and Objax☆127Updated last year
- ☆67Updated last year
- Swish Activation - PyTorch CUDA Implementation☆37Updated 5 years ago
- Implémentation of the article **Deep Learning CUDA Memory Usage and Pytorch optimization tricks**☆43Updated 5 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 3 years ago
- Train ImageNet in 18 minutes on AWS☆127Updated 10 months ago
- Toy implementations of some popular ML optimizers using Python/JAX☆43Updated 3 years ago
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification f…☆44Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago
- ☆70Updated 4 years ago
- Pretrained TorchVision models on CIFAR10 dataset (with weights)☆24Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆123Updated 6 months ago
- FLOPs and other statistics COunter for Pytorch neural networks☆23Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Pruning methods for pytorch with an optimizer-like interface☆15Updated 4 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆82Updated 3 years ago
- Visualization of various deep learning optimization algorithms using PyTorch automatic differentiation and optimizers.☆57Updated 6 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suite☆103Updated last year
- Unofficial pytorch implementation of ReZero in ResNet☆23Updated 4 years ago
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- TBA☆75Updated 5 years ago
- Implementation of Kronecker Attention in Pytorch☆18Updated 4 years ago
- All about acceleration and compression of Deep Neural Networks☆33Updated 5 years ago
- Pruning applied to Facial Recognition.☆15Updated 5 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- ☆38Updated 2 years ago
- ☆54Updated 4 years ago
- "Learning Rate Dropout" in PyTorch☆34Updated 5 years ago