gdalle / sparsediffaxLinks
Prototype for sparse automatic differentiation in JAX + Julia
☆18Updated 6 months ago
Alternatives and similar repositories for sparsediffax
Users that are interested in sparsediffax are comparing it to the libraries listed below
Sorting:
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆67Updated 3 weeks ago
- Numerical quadrature with JAX☆73Updated 2 weeks ago
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆17Updated 2 years ago
- Orthogonal polynomials with JAX☆26Updated last week
- ☆33Updated 5 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated 2 months ago
- A Review of Sensitivity Methods for Differential Equations☆33Updated last year
- Code for efficient solution of oscillatory ordinary differential equations☆32Updated 7 months ago
- Bilinear interpolation on grids with jax☆16Updated 3 years ago
- Interpolation and function approximation with JAX☆235Updated 2 weeks ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Updated last year
- Differentiable modelling on (multi)-GPUs workshop at JuliaCon23☆48Updated last year
- A fast direct solver for surface PDEs☆19Updated 2 weeks ago
- an encyclopedia of finite element definitions☆69Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆108Updated last week
- Efficient, Accurate, and Streamlined Training of Physics-Informed Neural Networks☆59Updated 9 months ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆108Updated this week
- A Julia library of summation-by-parts (SBP) operators used in finite difference, Fourier pseudospectral, continuous Galerkin, and discont…☆109Updated 3 weeks ago
- Analysis of initial value ODE solvers☆82Updated last year
- Python wrapper of LSODA (solving ODEs) which can be called from within numba functions.☆103Updated 2 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated last month
- Automatic Differentiation for Solid Mechanics☆58Updated last week
- A small library of paramaterizations and parameter constraints for PyTrees.☆47Updated 7 months ago
- High-level model-order reduction to automate the acceleration of large-scale simulations☆42Updated 2 weeks ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Updated 3 months ago
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated last year
- JAX + Units☆52Updated 2 weeks ago
- Notebook for my new solver book☆64Updated 2 months ago
- ☆36Updated 2 years ago
- Julia routines for working with MHD Equilibrium☆12Updated 8 months ago