Alternatives and similar repositories for random-walks:

Users that are interested in random-walks are comparing it to the libraries listed below

• cambridge-mlg / sgd-gp
  Public code for running Stochastic Gradient Descent on GPs.
  ☆36Updated 5 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆44Updated this week
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆33Updated last month
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆116Updated 2 months ago
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆34Updated last year
• 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 8 months ago
• danielward27 / flowjax
  ☆152Updated 3 weeks ago
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆31Updated last month
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆24Updated 11 months ago
• lollcat / fab-jax
  Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.
  ☆10Updated 9 months ago
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆71Updated 3 weeks ago
• bb515 / diffusionjax
  diffusionjax is a simple and accessible diffusion models package in JAX
  ☆41Updated 2 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆25Updated last month
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 3 weeks ago
• diffprog / diffprog.github.io
  Website for the book "The Elements of Differentiable Programming".
  ☆13Updated 6 months ago
• cgarciae / simple_flow_matching
  ☆18Updated 3 months ago
• DenisBless / variational_sampling_methods
  [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".
  ☆30Updated 3 months ago
• kazewong / flowMC
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆228Updated this week
• francois-rozet / inox
  Stainless neural networks in JAX
  ☆33Updated last week
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆53Updated this week
• geraschenko / gmrfs
  Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)
  ☆20Updated 11 months ago
• htlambley / functional_autoencoders
  Autoencoders in Function Space: Resolution-invariant autoencoders for functional data based on the functional variational autoencoder (FV…
  ☆12Updated 3 months 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 3 months ago
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆27Updated 4 months ago
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆16Updated last month
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆107Updated 2 weeks ago
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆165Updated this week
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆115Updated last month
• sail-sg / autofd
  Automatic Functional Differentiation in JAX
  ☆67Updated last week
• SaitejaUtpala / rieoptax
  Riemannian Optimization Using JAX
  ☆48Updated last year