A generic interface for linear algebra backends
☆78Mar 13, 2026Updated last week
Alternatives and similar repositories for lab
Users that are interested in lab are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Gaussian process modelling in Python☆226Dec 15, 2024Updated last year
- Kernels, the machine learning ones☆14Mar 6, 2023Updated 3 years ago
- Zonal Spherical Harmonics in d Dimensions in TensorFlow, PyTorch and Jax☆34Feb 10, 2026Updated last month
- Probabilistic inference for models of behaviour☆10Mar 5, 2026Updated 2 weeks ago
- A framework for composing Neural Processes in Python☆90Dec 17, 2024Updated last year
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 8 months ago
- Geometric kernels on manifolds, meshes and graphs☆561Mar 10, 2026Updated last week
- Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX☆23Mar 6, 2023Updated 3 years ago
- General framework for Bayesian inversion of continuous hierarchical models☆10Sep 20, 2021Updated 4 years ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- Gaussian processes in JAX and Flax.☆602Mar 14, 2026Updated last week
- Gaussian Process Factor Analysis with Dynamical Structure☆17Nov 3, 2020Updated 5 years ago
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- Multiple dispatch in Python☆638Updated this week
- Compositional Linear Algebra☆510Aug 1, 2025Updated 7 months ago
- Julia package to extend the conversion of Julia arrays to NumPy arrays☆15Jul 12, 2021Updated 4 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- Interactive textbook on state space models☆12Apr 11, 2022Updated 3 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆75Nov 13, 2020Updated 5 years ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆45Nov 6, 2023Updated 2 years ago
- PID-like control implemented as active inference with linear generative models☆11Jul 2, 2020Updated 5 years ago
- Tools for building equivariant polynomials on reductive Lie groups.☆36Jun 29, 2023Updated 2 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated 2 years ago
- Source for the sample efficient tabular RL submission to the 2019 NIPS workshop on Biological and Artificial RL☆24Apr 14, 2022Updated 3 years ago
- ☆18Oct 16, 2022Updated 3 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- A Bayesian optimization toolbox built on TensorFlow☆248Jan 14, 2026Updated 2 months ago
- Code of PriorVAE: encoding spatial priors with variational autoencoders☆12Jul 13, 2023Updated 2 years ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- GPS plugin for NetKet (www.netket.org), introducing new models, optimizers and Fermionic functionality.☆18Nov 24, 2025Updated 3 months ago
- ☆14Oct 7, 2022Updated 3 years ago
- A dead simple Python package for creating custom JAX pytree objects☆66Nov 14, 2024Updated last year
- Probabilistic Numerics in Python.☆459Jul 3, 2025Updated 8 months ago
- Code for the Neural Processes website and replication of 4 papers on NPs. Pytorch implementation.☆231Jun 12, 2024Updated last year
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,939Feb 23, 2026Updated last month
- (experimental) tracing and abstract interpretation with SymbolicUtils.jl☆22Mar 7, 2021Updated 5 years ago
- Ab-initio simulation of interacting fermions with equivariant normalizing flow.☆34Oct 23, 2022Updated 3 years ago