moskomule / shampoo.pytorch
An implementation of shampoo
☆74Updated 6 years ago
Alternatives and similar repositories for shampoo.pytorch:
Users that are interested in shampoo.pytorch are comparing it to the libraries listed below
- This repository is no longer maintained. Check☆81Updated 4 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- ☆34Updated 6 years ago
- Simple implementation of the LSUV initialization in PyTorch☆58Updated last year
- Implementation of "Learning with Random Learning Rates" in PyTorch.☆102Updated 5 years ago
- Code for "EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis" https://arxiv.org/abs/1905.05934☆112Updated 4 years ago
- Odds and Ends and Things I've implemented.☆78Updated 6 years ago
- The Singular Values of Convolutional Layers☆71Updated 6 years ago
- Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training o…☆148Updated 7 years ago
- A PyTorch implementation of shake-shake☆111Updated 4 years ago
- Utilities for Pytorch