pilancilab / scnn
Scalable Convex Neural Networks
☆23Updated last week
Alternatives and similar repositories for scnn:
Users that are interested in scnn are comparing it to the libraries listed below
- ☆13Updated 3 years ago
- ☆19Updated 2 years ago
- Riemannian Optimization Using JAX☆48Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆85Updated 3 years ago
- Flexible Inference for Predictive Coding Networks in JAX.☆32Updated 3 weeks ago
- The Energy Transformer block, in JAX☆56Updated last year
- Agustinus' very opiniated publication-ready plotting library☆62Updated 2 months ago
- ☆16Updated 6 months ago
- Code for the book "The Elements of Differentiable Programming".☆77Updated 2 weeks ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆39Updated 4 years ago
- JAX Arrays for human consumption☆90Updated last year
- Pre-trained Gaussian processes for Bayesian optimization☆90Updated 4 months ago
- Parameter-Free Optimizers for Pytorch☆122Updated 11 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆144Updated 5 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆43Updated last year
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- Turn jitted jax functions back into python source code☆22Updated 3 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆44Updated this week
- ☆15Updated 2 years ago
- Public code for running Stochastic Gradient Descent on GPs.☆36Updated 5 months ago
- Scalable training and inference for Probabilistic Circuits☆57Updated last month
- diffusionjax is a simple and accessible diffusion models package in JAX☆41Updated 2 months ago
- Bayesian model reduction for probabilistic machine learning☆10Updated last week
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- A generic interface for linear algebra backends☆73Updated last month
- ☆24Updated 2 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆37Updated 2 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆58Updated 9 months ago
- Neural Tangent Kernel (NTK) module for the scikit-learn library☆25Updated 5 months ago
- Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)☆20Updated 5 years ago