google-deepmind / eigengameLinks
Open source code for EigenGame.
☆30Updated 2 years ago
Alternatives and similar repositories for eigengame
Users that are interested in eigengame are comparing it to the libraries listed below
Sorting:
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- ☆115Updated this week
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…☆71Updated last month
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆180Updated last week
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- ☆192Updated last month
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62Updated 4 years ago
- JMP is a Mixed Precision library for JAX.☆207Updated 6 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- LoRA for arbitrary JAX models and functions☆140Updated last year
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Pytorch-like dataloaders for JAX.☆94Updated 2 months ago
- Differentiable Algorithms and Algorithmic Supervision.☆115Updated 2 years ago
- ☆232Updated 5 months ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Updated 2 years ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆22Updated 9 months ago
- Lightning-like training API for JAX with Flax☆42Updated 7 months ago
- Running Jax in PyTorch Lightning☆109Updated 7 months ago
- ☆74Updated 2 years ago
- ☆19Updated 3 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆150Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆130Updated 2 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated this week
- Supporing code for the paper "Bayesian Model Selection, the Marginal Likelihood, and Generalization".☆36Updated 3 years ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆188Updated 8 months ago
- A centralized place for deep thinking code and experiments☆85Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆107Updated last year