kach / gradient-descent-the-ultimate-optimizerLinks
Code for our NeurIPS 2022 paper
☆368Updated 2 years ago
Alternatives and similar repositories for gradient-descent-the-ultimate-optimizer
Users that are interested in gradient-descent-the-ultimate-optimizer are comparing it to the libraries listed below
Sorting:
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆331Updated last year
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 2 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆585Updated 5 months ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆276Updated 2 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆305Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,429Updated last week
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆479Updated 2 years ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated last year
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆207Updated last year
- Laplace approximations for Deep Learning.☆505Updated last month
- ☆775Updated last month
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆253Updated 2 months ago
- Tensors, for human consumption☆1,252Updated last week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆382Updated this week
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆438Updated 9 months ago
- Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"☆479Updated 2 years ago
- ☆228Updated 3 months ago
- Compositional Linear Algebra☆474Updated last week
- Constrained optimization toolkit for PyTorch☆678Updated 3 years ago
- D-Adaptation for SGD, Adam and AdaGrad☆519Updated 4 months ago
- Implementation of https://srush.github.io/annotated-s4☆495Updated 2 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 2 years ago
- Fast, differentiable sorting and ranking in PyTorch☆813Updated 3 weeks ago
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆378Updated last year
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆214Updated last week
- Optimal transport tools implemented with the JAX framework, to solve large scale matching problems of any flavor.☆600Updated 2 weeks ago
- Automatic gradient descent☆207Updated last year
- Fast Differentiable Sorting and Ranking☆604Updated last year