bremen79 / parameterfree
Parameter-Free Optimizers for Pytorch
☆109Updated 6 months ago
Related projects ⓘ
Alternatives and complementary repositories for parameterfree
- scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch☆18Updated 2 weeks ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆179Updated 3 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆107Updated this week
- Hessian spectral density estimation in TF and Jax☆115Updated 4 years ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- Agustinus' very opiniated publication-ready plotting library☆58Updated 2 months ago
- IVON optimizer for neural networks based on variational learning.☆53Updated 2 weeks ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆121Updated 2 years ago
- Simple (and cheap!) neural network uncertainty estimation☆55Updated 2 months ago
- Sketched matrix decompositions for PyTorch☆67Updated 6 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆127Updated last month
- Compositional Linear Algebra☆432Updated 3 weeks ago
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆76Updated 8 months ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆207Updated last month
- Laplace Redux -- Effortless Bayesian Deep Learning☆38Updated last year
- Large-scale, multi-GPU capable, kernel solver☆181Updated 4 months ago
- Pre-trained Gaussian processes for Bayesian optimization☆86Updated 2 months ago
- ☆97Updated 2 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆89Updated 7 months ago
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆58Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- ☆52Updated 3 months ago
- Riemannian Optimization Using JAX☆45Updated last year
- A library for uncertainty quantification based on PyTorch☆119Updated 2 years ago
- ☆235Updated last year
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 3 years ago
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆128Updated 5 years ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- Riemannian Convex Potential Maps☆68Updated last year
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆97Updated 4 years ago