gdalle / sparsediffaxLinks
Prototype for sparse automatic differentiation in JAX + Julia
☆18Updated 7 months ago
Alternatives and similar repositories for sparsediffax
Users that are interested in sparsediffax are comparing it to the libraries listed below
Sorting:
- Bilinear interpolation on grids with jax☆16Updated 3 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆68Updated last month
- Numerical quadrature with JAX☆75Updated last week
- 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
- Efficient, Accurate, and Streamlined Training of Physics-Informed Neural Networks☆59Updated 9 months ago
- Differentiable modelling on (multi)-GPUs workshop at JuliaCon23☆48Updated last year
- ☆36Updated 2 years ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆109Updated this week
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆17Updated 2 years ago
- Efficient Differentiable n-d PDE solvers in JAX.☆52Updated 2 months ago
- Orthogonal polynomials with JAX☆26Updated last week
- ☆34Updated 5 years ago
- An interface for accelerated simulation of high-dimensional collisionless and electrostatic plasmas.☆36Updated 3 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆53Updated 2 months ago
- A Review of Sensitivity Methods for Differential Equations☆33Updated last year
- Notebook for my new solver book☆64Updated 2 months ago
- Incompressible Navier-Stokes solver☆91Updated last month
- Interpolation and function approximation with JAX☆240Updated this week
- High-level model-order reduction to automate the acceleration of large-scale simulations☆42Updated last week
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆109Updated 2 weeks ago
- ETH course - Solving PDEs in parallel on GPUs I Fall 2025☆137Updated 2 weeks ago
- Almost trivial distributed parallelization of stencil-based GPU and CPU applications on a regular staggered grid☆192Updated last week
- Multivariate polynomial interpolation in Python. This is a mirror of the Minterpy repository. Please do not post issues or pull requests …☆39Updated last year
- A Julia package to construct orthogonal polynomials, their quadrature rules, and use it with polynomial chaos expansions.☆124Updated 2 weeks ago
- A Julia package for hierarchically semiseparable (HSS) matrices.☆26Updated 2 years ago
- Julia routines for working with MHD Equilibrium☆12Updated 8 months ago
- A scientific machine learning (SciML) wrapper for the FEniCS Finite Element library in the Julia programming language☆106Updated last week
- Probabilistic Numerical Differential Equation solvers via Bayesian filtering and smoothing☆131Updated last month
- Automatic-Differentiation-Enabled Plasma Transport in JAX☆35Updated last week
- This is a Julia package of nonlinear solvers. These codes are used in my book, Solving Nonlinear Equations with Iterative Methods: Solver…☆121Updated 3 weeks ago