bremen79 / parameterfree
Parameter-Free Optimizers for Pytorch
☆122Updated 11 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, ...)☆34Updated 3 weeks ago
- Agustinus' very opiniated publication-ready plotting library☆62Updated 2 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 8 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 11 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆171Updated this week
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆211Updated last week
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆112Updated this week
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- Sketched matrix decompositions for PyTorch☆70Updated 2 weeks ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- IVON optimizer for neural networks based on variational learning.☆60Updated 4 months ago
- Normalizing Flows using JAX☆83Updated last year
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆60Updated last year
- JAX Arrays for human consumption☆90Updated last year
- 🧱 Modula software package☆187Updated this week
- Hessian spectral density estimation in TF and Jax☆122Updated 4 years ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆129Updated 2 years ago
- PEPit is a package enabling computer-assisted worst-case analyses of first-order optimization methods.☆85Updated 2 months ago
- Simple (and cheap!) neural network uncertainty estimation☆62Updated last month
- Easy Hypernetworks in Pytorch and Jax☆99Updated 2 years ago
- Bayesianize: A Bayesian neural network wrapper in pytorch☆88Updated 10 months ago
- ☆53Updated 8 months ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆106Updated 4 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Image augmentation library for Jax☆39Updated 11 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆43Updated last year
- ☆245Updated 5 months ago
- Materials of the Nordic Probabilistic AI School 2022.☆175Updated 2 years ago
- Pre-trained Gaussian processes for Bayesian optimization☆90Updated 4 months ago