cooper-org / cooper
A general-purpose, deep learning-first library for constrained optimization in PyTorch
☆107Updated this week
Related projects ⓘ
Alternatives and complementary repositories for cooper
- Agustinus' very opiniated publication-ready plotting library☆58Updated 2 months ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- Code for the article "What if Neural Networks had SVDs?", to be presented as a spotlight paper at NeurIPS 2020.☆69Updated 3 months ago
- Normalizing Flows using JAX☆82Updated 11 months ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆170Updated 2 years ago
- IVON optimizer for neural networks based on variational learning.☆53Updated 2 weeks ago
- ☆145Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆109Updated 6 months ago
- scipy linear operators for the Hessian, Fisher/GGN, and more in PyTorch☆18Updated 2 weeks ago
- Riemannian Optimization Using JAX☆45Updated last year
- The Modified Differential Multiplier Method (MDMM) for PyTorch☆51Updated 3 years ago
- Hessian spectral density estimation in TF and Jax☆115Updated 4 years ago
- Simple (and cheap!) neural network uncertainty estimation☆55Updated 2 months ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆95Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆38Updated last year
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 3 years ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆89Updated 7 months ago
- ☆49Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆96Updated last year
- Normalizing Flows with a resampled base distribution☆44Updated 2 years ago
- Manifold-learning flows (ℳ-flows)☆230Updated 4 years ago
- Pytorch implementation of the Power Spherical distribution☆73Updated 4 months ago
- Regularized Neural ODEs (RNODE)☆82Updated 3 years ago
- TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆76Updated 5 months ago
- Pre-trained Gaussian processes for Bayesian optimization☆86Updated 2 months ago
- Bayesian algorithm execution (BAX)☆48Updated 3 years ago
- Convex potential flows☆78Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆37Updated 4 years ago