Alternatives and similar repositories for jaxgeometry:

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

• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆55Updated 3 months ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆116Updated 2 months ago
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆24Updated 11 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…
  ☆20Updated 8 months ago
• CUQI-DTU / CUQIpy
  ☆48Updated last week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆165Updated last week
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆53Updated last week
• Ceyron / pdequinox
  Neural Emulator Architectures in JAX.
  ☆16Updated 4 months ago
• laplax-org / laplax
  Laplace approximations in JAX.
  ☆16Updated last week
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆53Updated 3 years ago
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆27Updated 4 months ago
• yifanc96 / NonLinPDEs-GPsolver
  Code for the paper: Solving and Learning Nonlinear PDEs with Gaussian Processes
  ☆36Updated 5 months ago
• eepperly / XTrace
  Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"
  ☆13Updated last year
• timothyas / oden-beamer-template
  Oden Institute template for beamer users
  ☆12Updated 4 years ago
• cornellius-gp / linear_operator
  A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch
  ☆106Updated 2 weeks ago
• PredictiveScienceLab / advanced-scientific-machine-learning
  ME 697 - Advanced Scientific Machine Learning
  ☆21Updated this week
• BallisticLA / parla
  Python Algorithms for Randomized Linear Algebra
  ☆53Updated last year
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆93Updated last year
• yifanc96 / GaussianProcessPDEs
  An integrated demo: Gaussian processes for PDEs and inverse problems
  ☆13Updated 2 months ago
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆15Updated last week
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆16Updated 3 weeks ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆89Updated 3 weeks ago
• probabilistic-numerics / probnum-spring-school
  Tutorial materials of the Probabilistic Numerics Spring School.
  ☆34Updated last year
• numpi / hm-toolbox
  Toolbox for HODLR and HSS matrices in MATLAB
  ☆45Updated last year
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆41Updated 3 weeks ago
• scalable-matrix / H2Pack
  H2 Matrix Package
  ☆29Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆123Updated 6 months ago
• tum-pbs / apebench
  [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…
  ☆64Updated 4 months ago
• Dipolar-Quantum-Gases / jaxfit
  GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
  ☆56Updated last year
• rfeinman / Torch-ARPACK
  Fast extremal eigensolvers for PyTorch.
  ☆17Updated 3 years ago