pilancilab / scnn
Scalable Convex Neural Networks
☆22Updated 3 months ago
Alternatives and similar repositories for scnn:
Users that are interested in scnn are comparing it to the libraries listed below
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- Flexible Inference for Predictive Coding Networks in JAX.☆31Updated last month
- ☆16Updated 5 months ago
- Tools for JAX☆45Updated last week
- ☆13Updated 3 years ago
- Bayesian model reduction for probabilistic machine learning☆10Updated 2 months ago
- ☆19Updated 2 years ago
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- The Energy Transformer block, in JAX☆56Updated last year
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆142Updated 4 months ago
- ☆15Updated 2 years ago
- A PyTorch library for all things nonlinear control and reinforcement learning.☆44Updated 3 years ago
- ☆39Updated last year
- Riemannian Optimization Using JAX☆48Updated last year
- Code for the book "The Elements of Differentiable Programming".☆73Updated last week
- Parameter-Free Optimizers for Pytorch☆113Updated 9 months ago
- Recursive Bayesian Estimation (Sequential / Online Inference)☆58Updated 10 months ago
- JAX Arrays for human consumption☆90Updated last year
- Pre-trained Gaussian processes for Bayesian optimization☆90Updated 2 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆41Updated 3 months ago
- Agustinus' very opiniated publication-ready plotting library☆61Updated 2 weeks ago
- PyTorch implementation for "Probabilistic Circuits for Variational Inference in Discrete Graphical Models", NeurIPS 2020☆16Updated 3 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆43Updated last month
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆19Updated 4 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆41Updated last year
- Exponential families for JAX☆63Updated last week
- Simple (and cheap!) neural network uncertainty estimation☆61Updated last week
- Neuroevolution Benchmark in JAX 🦕☆38Updated last year
- A generic interface for linear algebra backends☆71Updated 7 months ago