ucl-bug / jaxdf
A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
☆121Updated 2 months ago
Related projects ⓘ
Alternatives and complementary repositories for jaxdf
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆89Updated last year
- Differentiable interface to FEniCS for JAX☆50Updated 3 years ago
- Stencil computations in JAX☆66Updated last year
- Interpolation and function approximation with JAX☆135Updated 3 weeks ago
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆94Updated 2 months ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆80Updated last week
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆34Updated 3 weeks ago
- Multiple dispatch over abstract array types in JAX.☆105Updated last week
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆41Updated last month
- ☆107Updated 3 weeks ago
- Mathematical operations for JAX pytrees☆189Updated 6 months ago
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆75Updated 2 weeks ago
- Minimal Implementation of Bayesian Optimization in JAX☆85Updated 6 months ago
- A dead simple Python package for creating custom JAX pytree objects☆62Updated last week
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆53Updated 2 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆96Updated 3 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆61Updated 4 months ago
- Code for the book "The Elements of Differentiable Programming".☆65Updated 3 months ago
- Exponential families for JAX☆55Updated this week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆96Updated last year
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Learning function operators with neural networks.☆25Updated 2 months ago
- GPU/TPU accelerated nonlinear least-squares curve fitting using JAX☆52Updated last year
- A generic interface for linear algebra backends☆70Updated 4 months ago
- JAX Arrays for human consumption☆88Updated last year
- Bootcamp notebooks☆50Updated 6 months ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆21Updated 6 months ago
- Tools for JAX☆43Updated this week
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆38Updated last week