bremen79 / parameterfree
Parameter-Free Optimizers for Pytorch
☆120Updated 10 months ago
Alternatives and similar repositories for parameterfree:
Users that are interested in parameterfree are comparing it to the libraries listed below
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆32Updated this week
- Agustinus' very opiniated publication-ready plotting library☆61Updated last month
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆112Updated 2 weeks ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 10 months ago
- Riemannian Optimization Using JAX☆48Updated last year
- Simple (and cheap!) neural network uncertainty estimation☆61Updated 2 weeks ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆183Updated 2 months ago
- IVON optimizer for neural networks based on variational learning.☆59Updated 3 months ago
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆211Updated this week
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 2 years ago
- Normalizing Flows using JAX☆82Updated last year
- Hessian spectral density estimation in TF and Jax☆121Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆107Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆168Updated 2 months ago
- Pytorch-like dataloaders for JAX.☆75Updated 4 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆41Updated last year
- PEPit is a package enabling computer-assisted worst-case analyses of first-order optimization methods.☆85Updated last month
- A Python package of computer vision models for the Equinox ecosystem.☆104Updated 7 months ago
- Materials of the Nordic Probabilistic AI School 2023.☆89Updated last year
- Euclidean Wasserstein-2 optimal transportation☆45Updated last year
- JAX Arrays for human consumption☆90Updated last year
- A primer on Bayesian Neural Networks. The aim of this reading list is to facilitate the entry of new researchers into the field of Bayesi…☆52Updated last year
- Pytorch implementation of KFAC and E-KFAC (Natural Gradient).☆130Updated 5 years ago
- A functional training loops library for JAX☆86Updated last year
- ☆53Updated 7 months ago
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆59Updated last year
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated 9 months ago
- ☆65Updated 2 months ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆114Updated last week