davda54 / ada-hessian
Easy-to-use AdaHessian optimizer (PyTorch)
β77Updated 4 years ago
Alternatives and similar repositories for ada-hessian:
Users that are interested in ada-hessian are comparing it to the libraries listed below
- π§ Pytorch code for the Fromage optimiser.β123Updated 6 months ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learningβ270Updated last year
- PyTorch-SSO: Scalable Second-Order methods in PyTorchβ144Updated last year
- Structured matrices for compressing neural networksβ66Updated last year
- β98Updated 3 years ago
- Fast Discounted Cumulative Sums in PyTorchβ95Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β106Updated 2 years ago
- Official code repository of the paper Linear Transformers Are Secretly Fast Weight Programmers.β102Updated 3 years ago
- DeepOBS: A Deep Learning Optimizer Benchmark Suiteβ103Updated last year
- Very deep VAEs in JAX/Flaxβ46Updated 3 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β154Updated last month
- ASDL: Automatic Second-order Differentiation Library for PyTorchβ182Updated last month
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.β72Updated 6 months ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitableβ211Updated 3 years ago
- π© Pytorch and Jax code for the Madam optimiser.β51Updated 3 years ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"β180Updated 3 years ago
- β36Updated 2 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilitiesβ209Updated 9 months ago
- PyTorch implementation of HashedNetsβ36Updated last year
- Block-sparse primitives for PyTorchβ154Updated 3 years ago
- PadΓ© Activation Units: End-to-end Learning of Activation Functions in Deep Neural Networkβ63Updated 3 years ago
- A GPT, made only of MLPs, in Jaxβ57Updated 3 years ago
- Reparameterize your PyTorch modulesβ70Updated 4 years ago
- β67Updated last year
- Hessian trace estimation using PyTorch and Hutch++β19Updated 4 years ago
- Code base for SRSGD.β28Updated 4 years ago
- Monotone operator equilibrium networksβ51Updated 4 years ago
- Butterfly matrix multiplication in PyTorchβ165Updated last year
- A pytorch implementation for the LSTM experiments in the paper: Why Gradient Clipping Accelerates Training: A Theoretical Justification fβ¦β44Updated 4 years ago
- a lightweight transformer library for PyTorchβ72Updated 3 years ago