Alternatives and similar repositories for 4M24

Users that are interested in 4M24 are comparing it to the libraries listed below

• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆141Updated 2 weeks ago
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 7 months ago
• alonfnt / pcax
  Differentiable Principal Component Analysis (PCA) implementation in JAX
  ☆29Updated 5 months ago
• sbi-benchmark / sbibm
  Simulation-based inference benchmark
  ☆102Updated 8 months ago
• pnkraemer / tornadox
  Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…
  ☆21Updated last year
• j-wilson / GPflowSampling
  Code for efficiently sampling functions from GP(flow) posteriors
  ☆73Updated 4 years ago
• ludwigwinkler / pytorch_MCMC
  Lightweight MCMC sampling for PyTorch Models aka My Corona Project
  ☆49Updated 2 months ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆47Updated 8 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆53Updated 3 weeks ago
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆35Updated 7 months ago
• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆39Updated 5 months ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆115Updated last week
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆97Updated 5 months ago
• jeremiecoullon / SGMCMCJax
  Lightweight library of stochastic gradient MCMC algorithms written in JAX.
  ☆105Updated last year
• wiseodd / pub-ready-plots
  Agustinus' very opiniated publication-ready plotting library
  ☆69Updated 5 months ago
• jeremiecoullon / jax-tqdm
  Add a tqdm progress bar to your JAX scans and loops.
  ☆117Updated 5 months ago
• danielward27 / flowjax
  ☆189Updated last month
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆35Updated 2 years ago
• diffprog / diffprog.github.io
  Website for the book "The Elements of Differentiable Programming".
  ☆14Updated 3 months ago
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆128Updated 3 weeks ago
• jamesvuc / jax-bayes
  Bayesian inference with Python and Jax.
  ☆34Updated 2 years ago
• Information-Fusion-Lab-Umass / NuX
  Normalizing Flows using JAX
  ☆84Updated last year
• philipphennig / NumericsOfML
  Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen
  ☆218Updated last year
• ThGaskin / NeuralABM
  Neural parameter calibration for multi-agent models. Uses neural networks to estimate marginal densities on parameters and networks
  ☆31Updated last month
• AaltoML / BayesNewton
  Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.
  ☆237Updated last year
• jhuggins / viabel
  Efficient, lightweight variational inference and approximation bounds
  ☆45Updated last year
• dodgebc / jaxkd
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆50Updated 2 months ago
• hd-UQ / cd_dynamax
  Extension of dynamax repo to cases with continuous-time dynamics with measurements sampled at possibly irregular discrete times. Allows …
  ☆14Updated 3 months ago
• fidero / gp-derivative
  Example code to speed up GP inference with gradients for high-dimensional inputs.
  ☆14Updated 4 years ago
• EEA-sensors / sqrt-parallel-smoothers
  A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…
  ☆12Updated 10 months ago