Alternatives and similar repositories for laplax

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

• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆107Updated 2 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆118Updated 4 months ago
• andres-fr / skerch
  Sketched matrix decompositions for PyTorch
  ☆70Updated 3 weeks ago
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆34Updated 2 years ago
• modichirag / GSM-VI
  Code for Gaussian Score Matching Variational Inference
  ☆33Updated 2 months ago
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆26Updated last year
• ComputationalEvolutionaryMorphometry / jaxgeometry
  ☆13Updated 8 months ago
• ludwigwinkler / pytorch_MCMC
  Lightweight MCMC sampling for PyTorch Models aka My Corona Project
  ☆48Updated 4 years ago
• f-dangel / curvlinops
  PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)
  ☆38Updated 3 weeks ago
• danielward27 / flowjax
  ☆160Updated last week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆182Updated last week
• team-approx-bayes / ivon
  IVON optimizer for neural networks based on variational learning.
  ☆65Updated 6 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 5 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆46Updated last month
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆58Updated last week
• wesselb / lab
  A generic interface for linear algebra backends
  ☆73Updated 2 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 9 months ago
• bmucsanyi / untangle
  Large-scale uncertainty benchmark in deep learning.
  ☆57Updated last week
• EddieCunningham / generax
  Generative models with JAX.
  ☆17Updated 4 months ago
• JonathanWenger / itergp
  IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)
  ☆41Updated 2 years ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆57Updated 5 months ago
• JaxGaussianProcesses / GPJax
  Gaussian processes in JAX and Flax.
  ☆507Updated last month
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆121Updated 2 weeks ago
• kazewong / flowMC
  Normalizing-flow enhanced sampling package for probabilistic inference in Jax
  ☆233Updated this week
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆73Updated 2 weeks ago
• ylefay / bayesianSDEsolver
  Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.
  ☆10Updated 3 months ago
• MachineLearningLifeScience / stochman
  Algorithms for computations on random manifolds made easier
  ☆89Updated last year
• google / tree-math
  Mathematical operations for JAX pytrees
  ☆199Updated 5 months ago
• patrick-kidger / optimistix
  Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/
  ☆413Updated this week
• probabilists / lampe
  Likelihood-free AMortized Posterior Estimation with PyTorch
  ☆127Updated 8 months ago