adam-coogan / jaxinterp2d

Related projects ⓘ

Alternatives and complementary repositories for jaxinterp2d

• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆47Updated 9 months ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆44Updated 3 weeks ago
• NIFTy-PPL / JAXbind
  Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation
  ☆61Updated 4 months ago
• DENG-MIT / StiffNeuralODE
  Stiff Neural Ordinary Differential Equations
  ☆30Updated last year
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆135Updated 3 weeks ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆50Updated 3 years ago
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆10Updated 3 weeks ago
• flatironinstitute / jax-finufft
  JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library
  ☆80Updated last week
• b03901165Shih / 2D_FPE_Solver
  Two Dimensional Fokker-Planck Solver using Matlab
  ☆17Updated 5 years ago
• yifanc96 / GaussianProcessPDEs
  An integrated demo: Gaussian processes for PDEs and inverse problems
  ☆13Updated 5 months ago
• lukasheinrich / pyhep2020-autodiff-tutorial
  ☆31Updated 4 years ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆34Updated 3 weeks ago
• JakobRobnik / MicroCanonicalHMC
  MCHMC: sampler from an arbitrary differentiable distribution
  ☆70Updated this week
• AshKelly / pyquad
  A lightweight, parallel python wrapper for the GSL integration library
  ☆13Updated 3 years ago
• RehMoritz / vmc_pde
  Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…
  ☆9Updated 2 years ago
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆24Updated 3 weeks ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆42Updated 3 years ago
• Nicholaswogan / NumbaMinpack
  Python wrapper of Minpack (root finding) which can be called from within numba functions.
  ☆32Updated 2 years ago
• flaport / klujax
  Solve sparse linear systems in JAX using the KLU algorithm
  ☆33Updated last week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆121Updated 2 months ago
• dfm / jpu
  JAX + Units
  ☆43Updated last week
• ScientificComputingCWI / SemesterProgramme-SciML
  Material for workshop and autumn school on scientific machine learning 2023
  ☆19Updated 11 months ago
• NIFTy-PPL / NIFTy
  Probabilistic programming framework for signal inference algorithms that operate regardless of the underlying grids and their resolutions
  ☆21Updated this week
• HajimeKawahara / exojax
  🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt
  ☆57Updated this week
• Nicholaswogan / numbalsoda
  Python wrapper of LSODA (solving ODEs) which can be called from within numba functions.
  ☆92Updated last year
• ergodicio / adept
  Automatic-Differentiation-Enabled Plasma Transport in JAX
  ☆27Updated this week
• Dipolar-Quantum-Gases / jaxfit
  GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
  ☆52Updated last year
• pmocz / finitevolume-jax
  Compressible Euler equations solved with finite volume implemented in JAX, plugged into an optimization loop
  ☆17Updated 5 months ago
• moble / spherical
  Evaluate and transform D matrices, 3-j symbols, and (scalar or spin-weighted) spherical harmonics
  ☆45Updated 11 months ago
• johnaparker / fplanck
  Numerically solve the Fokker-Planck equation in N dimensions
  ☆82Updated last year