☆314Feb 11, 2026Updated 2 weeks ago
Alternatives and similar repositories for enn
Users that are interested in enn are comparing it to the libraries listed below
Sorting:
- ☆31Aug 25, 2022Updated 3 years ago
- ☆195Jan 9, 2026Updated last month
- Train very large language models in Jax.☆210Oct 21, 2023Updated 2 years ago
- High-quality implementations of standard and SOTA methods on a variety of tasks.☆1,567Feb 2, 2026Updated last month
- ☆620Jan 23, 2026Updated last month
- ☆1,407Dec 9, 2025Updated 2 months ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Oct 5, 2021Updated 4 years ago
- Official Implementation of "Natural Posterior Network: Deep Bayesian Predictive Uncertainty for Exponential Family Distributions" (ICLR, …☆87Apr 5, 2023Updated 2 years ago
- JAX-based neural network library☆3,191Feb 24, 2026Updated last week
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 5 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆63Jun 17, 2024Updated last year
- ☆330Aug 22, 2023Updated 2 years ago
- ☆164Dec 13, 2023Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.☆434Mar 6, 2025Updated 11 months ago
- CLU lets you write beautiful training loops in JAX.☆367Updated this week
- Riemannian Convex Potential Maps☆67Mar 31, 2023Updated 2 years ago
- Official Implementation of "Transformers Can Do Bayesian Inference", the PFN paper☆253Oct 24, 2024Updated last year
- Contrastive Language-Image Pretraining☆144Sep 6, 2022Updated 3 years ago
- Uncertainty Toolbox: a Python toolbox for predictive uncertainty quantification, calibration, metrics, and visualization☆1,977Mar 5, 2025Updated 11 months ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Oct 28, 2022Updated 3 years ago
- Exponential families for JAX☆76Feb 20, 2026Updated last week
- Laplace approximations for Deep Learning.☆535Apr 22, 2025Updated 10 months ago
- A repo based on XiLin Li's PSGD repo that extends some of the experiments.☆14Oct 7, 2024Updated last year
- Code for "Training, Architecture, and Prior for Deterministic Uncertainty Methods" ICLR 2023 Workshop on Trustworthy ML☆12Jun 15, 2023Updated 2 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 3 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,021Feb 3, 2026Updated last month
- ☆774Jan 27, 2024Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,792Feb 23, 2026Updated last week
- Fast and Easy Infinite Neural Networks in Python☆2,375Mar 1, 2024Updated 2 years ago
- PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks☆467Aug 28, 2024Updated last year
- code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"☆24Dec 29, 2021Updated 4 years ago
- Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-Counts (Neurips 2020)☆78May 18, 2022Updated 3 years ago
- ☆120Jan 27, 2026Updated last month
- A Graph Neural Network Library in Jax☆1,465Mar 18, 2024Updated last year
- A repository with implementations of major papers on Gaussian Process regression models, implemented from scratch in Python, notably incl…☆14Nov 19, 2022Updated 3 years ago
- Starbucks: Improved Training for 2D Matryoshka Embeddings☆22Jun 30, 2025Updated 8 months ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,529Apr 13, 2024Updated last year