Finite difference tools in JAX
☆23Feb 4, 2024Updated 2 years ago
Alternatives and similar repositories for finitediffX
Users that are interested in finitediffX are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Orthogonal polynomials with JAX☆32Updated this week
- Special functions in Jax☆20Updated this week
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆48Aug 18, 2026Updated 2 months ago
- Can we make a usable XSPEC model module in Python **experimental**☆14Oct 10, 2024Updated 2 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆83Feb 12, 2026Updated 8 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Matrix-free linear algebra in JAX.☆183Oct 22, 2026Updated last week
- JAX + Units☆54Mar 2, 2026Updated 8 months ago
- Unitful Quantities in JAX☆70Updated this week
- Probabilistic Programming and Nested sampling in JAX☆262Oct 27, 2026Updated last week
- Training methodologies for autoregressive neural operators/emulators in JAX.☆16Jun 1, 2026Updated 5 months ago
- A JAX-based gaussian process structure learning package for time series modelling.☆37Apr 28, 2026Updated 6 months ago
- A fast GPU based bispectrum estimator implemented using JAX☆12Sep 14, 2026Updated last month
- Efficient library for spectral analysis in high-energy astrophysics.☆24Oct 26, 2026Updated last week
- A modular X-ray spectral-timing astronomy modelling package☆23Updated this week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Powerful shallow-water implementations in pure Python☆18Aug 18, 2024Updated 2 years ago
- Python interface to the QDLDL (https://github.com/osqp/qdldl) free LDL factorization routine for quasi-definite linear systems☆18Apr 6, 2026Updated 6 months ago
- Fast, lightweight and parallelised simulation-based inference in JAX.☆24Oct 28, 2025Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- ☆12Aug 26, 2025Updated last year
- A scalable MPI library for computing fast Fourier transforms in python.☆11Oct 12, 2026Updated 3 weeks ago
- Python library implementing the Levin method to calculate integrals containing products of up to three Bessel functions☆14Oct 21, 2026Updated last week
- Fisher forecasting code for use with class_delens☆13Mar 31, 2026Updated 7 months ago
- Solve the 2D shallow water equations with a finite volume method based on a Q-scheme☆14Mar 30, 2023Updated 3 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Interpolation and function approximation with JAX☆283Updated this week
- Agent-assisted hyperparameter optimization 📈☆17Aug 2, 2026Updated 3 months ago
- Add a tqdm progress bar to your JAX scans and loops.☆131Updated this week
- Multiple dispatch over abstract array types in JAX.☆143Oct 21, 2026Updated 2 weeks ago
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆144Oct 21, 2026Updated last week
- Code for measuring bispectra from cosmological simulation snapshots☆12Mar 17, 2020Updated 6 years ago
- A differentiable cosmology library in JAX☆248Jun 27, 2025Updated last year
- A dataset and benchmark for emulating exoplanet climates☆20Sep 25, 2026Updated last month
- Use numba-compiled kernels from within Jax☆30May 1, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Efficient Differentiable n-d PDE Solvers in JAX.☆233Updated this week
- Public branch of CONNECT☆14Aug 25, 2025Updated last year
- Lightning-like training API for JAX with Flax☆45Dec 8, 2024Updated last year
- ☆18Aug 18, 2021Updated 5 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆587Oct 27, 2026Updated last week
- Differentiable Principal Component Analysis (PCA) implementation in JAX☆36Sep 28, 2026Updated last month
- UnDiFi-2D: an Unstructured Discontinuity Fitting code for 2D grids.☆11Oct 5, 2026Updated 3 weeks ago