bremen79 / parameterfree
Parameter-Free Optimizers for Pytorch
☆109Updated 9 months ago
Alternatives and similar repositories for parameterfree:
Users that are interested in parameterfree are comparing it to the libraries listed below
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆111Updated this week
- CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.☆77Updated 10 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆103Updated 6 months ago
- Tutorial on amortized optimization for learning to optimize over continuous domains☆236Updated last year
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆58Updated last year
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆97Updated 9 months ago
- ☆98Updated 3 years ago
- IVON optimizer for neural networks based on variational learning.☆58Updated 2 months ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆128Updated 2 years ago
- ☆15Updated 4 months ago
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆25Updated last week
- Riemannian Optimization Using JAX☆48Updated last year
- {KFAC,EKFAC,Diagonal,Implicit} Fisher Matrices and finite width NTKs in PyTorch☆210Updated 3 months ago
- Code for the paper: "Tensor Programs II: Neural Tangent Kernel for Any Architecture"☆103Updated 4 years ago
- Hessian spectral density estimation in TF and Jax☆120Updated 4 years ago
- Simple (and cheap!) neural network uncertainty estimation☆60Updated this week
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆182Updated last month
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆154Updated last month
- Agustinus' very opiniated publication-ready plotting library☆61Updated 4 months ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆59Updated 3 years ago
- ☆63Updated last month
- Large-scale, multi-GPU capable, kernel solver☆184Updated 6 months ago
- ☆53Updated 6 months ago
- Algorithms for computations on random manifolds made easier☆87Updated last year
- ☆67Updated 5 years ago
- JAX Arrays for human consumption☆90Updated last year
- ☆191Updated 8 months ago
- ☆35Updated 2 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆41Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago