pilancilab / scnnLinks
Scalable Convex Neural Networks
☆24Updated 4 months ago
Alternatives and similar repositories for scnn
Users that are interested in scnn are comparing it to the libraries listed below
Sorting:
- Loopy belief propagation for factor graphs on discrete variables in JAX☆156Updated 11 months ago
- The Energy Transformer block, in JAX☆59Updated last year
- ☆16Updated 4 years ago
- Riemannian Optimization Using JAX☆52Updated last year
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Official PyTorch implementation of NeuralSVD (ICML 2024)☆20Updated last year
- ☆32Updated 11 months ago
- ☆54Updated 3 months ago
- Parallelizing non-linear sequential models over the sequence length☆54Updated 2 months ago
- Open source code for EigenGame.☆30Updated 2 years ago
- ☆35Updated 9 months ago
- ☆19Updated 3 years ago
- Recursive Bayesian Estimation (Sequential / Online Inference)☆59Updated last year
- Flexible Inference for Predictive Coding Networks in JAX.☆53Updated this week
- Scalable training and inference for Probabilistic Circuits☆75Updated 2 months ago
- ☆15Updated 3 years ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆139Updated 3 months ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆130Updated 3 years ago
- Stochastic Automatic Differentiation library for PyTorch.☆206Updated last year
- Bayesian inference with Python and Jax.☆34Updated 2 years ago
- Agustinus' very opiniated publication-ready plotting library☆69Updated 4 months ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Repo for the paper "Landscape Surrogate Learning Decision Losses for Mathematical Optimization Under Partial Information"☆36Updated 2 years ago
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆89Updated 3 years ago
- About A collection of AWESOME things about information geometry Topics☆165Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆22Updated last year
- Public code for running Stochastic Gradient Descent on GPs.☆39Updated 4 months ago
- A Python package of computer vision models for the Equinox ecosystem.☆109Updated last year
- Probabilistic programming with large language models☆135Updated last month