damek / gd-lean
☆19Updated last year
Alternatives and similar repositories for gd-lean:
Users that are interested in gd-lean are comparing it to the libraries listed below
- ☆51Updated last month
- Riemannian Optimization Using JAX☆49Updated last year
- PEPit is a package enabling computer-assisted worst-case analyses of first-order optimization methods.☆86Updated 3 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- ☆80Updated 3 years ago
- Code of the Performance Estimation Toolbox (PESTO) whose aim is to ease the access to the PEP methodology for performing worst-case analy…☆54Updated last year
- A matlab implementation for sampling log-concave distributions with polytope constraints☆16Updated 7 months ago
- Code for the book "The Elements of Differentiable Programming".☆84Updated last month
- Differentiation through cone programs☆99Updated 3 months ago
- Turn jitted jax functions back into python source code☆22Updated 4 months ago
- ☆67Updated 5 months ago
- Inference Combinators in JAX☆47Updated 6 months ago
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- Parameter-Free Optimizers for Pytorch☆126Updated last year
- MATLAB implementation of the Scalable Semidefinite Programming☆26Updated last year
- Exponential families for JAX☆69Updated this week
- Einsum-like high-level array sharding API for JAX☆34Updated 9 months ago
- Multiple dispatch over abstract array types in JAX.☆120Updated this week
- Scalable Convex Neural Networks☆24Updated last week
- Tools for JAX☆46Updated 3 weeks ago
- Fast hyperparameter settings for non-smooth estimators:☆40Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆150Updated 6 months ago
- ☆12Updated 3 years ago
- Pytrees + dataclasses ❤️☆62Updated last week
- Automatic Functional Differentiation in JAX☆72Updated last month
- Code for the paper "Pretraining task diversity and the emergence of non-Bayesian in-context learning for regression"☆21Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆37Updated 2 weeks ago
- ☆11Updated 2 weeks ago
- ☆139Updated 2 weeks ago
- ☆21Updated 2 years ago