pilancilab / scnnLinks
Scalable Convex Neural Networks
☆24Updated 2 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☆154Updated 8 months ago
- The Energy Transformer block, in JAX☆57Updated last year
- Agustinus' very opiniated publication-ready plotting library☆67Updated 2 months ago
- ☆17Updated 10 months ago
- ☆49Updated last month
- Parameter-Free Optimizers for Pytorch☆130Updated last year
- Riemannian Optimization Using JAX☆49Updated last year
- ☆15Updated 4 years ago
- Pre-trained Gaussian processes for Bayesian optimization☆93Updated 3 months ago
- Public code for running Stochastic Gradient Descent on GPs.☆39Updated 2 months ago
- Flexible Inference for Predictive Coding Networks in JAX.☆51Updated 2 weeks ago
- This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…☆130Updated 2 years ago
- Recursive Bayesian Estimation (Sequential / Online Inference)☆59Updated last year
- ☆15Updated 2 years ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 6 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Code for "Bayesian Structure Learning with Generative Flow Networks"☆87Updated 3 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated last month
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆40Updated 4 years ago
- Bayesian inference with Python and Jax.☆32Updated 2 years ago
- This repository contains PyTorch implementations of various random feature maps for dot product kernels.☆21Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆84Updated last year
- Code for experiments on transformers using Markovian data.☆17Updated 7 months ago
- Hierarchical Associative Memory User Experience☆101Updated last year
- Bayesian active learning with EPIG data acquisition☆32Updated 2 months ago
- Repo for the paper "Landscape Surrogate Learning Decision Losses for Mathematical Optimization Under Partial Information"☆36Updated last year
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆53Updated 8 months ago
- A general-purpose, deep learning-first library for constrained optimization in PyTorch☆129Updated last month
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆29Updated last year
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated last year