Efficient forward- and reverse-mode sparse Jacobians using Jax
☆70Dec 1, 2025Updated 3 months ago
Alternatives and similar repositories for sparsejac
Users that are interested in sparsejac are comparing it to the libraries listed below
Sorting:
- Solve sparse linear systems in JAX using the KLU algorithm☆38Feb 19, 2026Updated last month
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 5 months ago
- measure minimum solid/void lengthscales in binary image☆21Dec 8, 2025Updated 3 months ago
- testbed problems for photonics optimization☆46Jun 22, 2024Updated last year
- A package includes various time-domain numerical solvers for the Maxwell's equations.☆14Jan 18, 2024Updated 2 years ago
- pjz is JAX and fdtd-z -- scalable photonic simulation and optimization☆21Mar 27, 2024Updated last year
- Real-time design of architectural structures with differentiable mechanics and neural networks [ICLR 2025]☆23Jun 13, 2025Updated 9 months ago
- ☆35Jul 19, 2023Updated 2 years ago
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Feb 8, 2025Updated last year
- Fourier Modal Method with Jax (official maintained repo)☆17Oct 21, 2025Updated 4 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆50Sep 14, 2024Updated last year
- geometric primitives for piecewise functions on grids☆24Jun 28, 2024Updated last year
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆526Mar 10, 2026Updated last week
- Vector Finite Difference Maxwell Modesolver☆16Jun 8, 2024Updated last year
- A GPU-accelerated, differentiable, Maxwell FDTD solver☆72Feb 23, 2026Updated 3 weeks ago
- A differentiable finite element analysis solver for structural optimization based on JAX☆40Dec 22, 2025Updated 2 months ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆165Mar 2, 2026Updated 2 weeks ago
- Dataset of solutions to inverse design challenges☆16Oct 23, 2025Updated 4 months ago
- Artificial nanofabrication of integrated photonic circuits using deep learning☆72Feb 4, 2026Updated last month
- Code accompanying the NeurIPS 2021 Paper: A Probabilistic State Space Model for Joint Inference from Differential Equations and Data (Sch…☆13Nov 7, 2022Updated 3 years ago
- Easily mesh collections of 2D and 3D polygons with complex extrusions using GMSH☆29Updated this week
- ☆45Oct 3, 2022Updated 3 years ago
- Auto-differentiable and hardware-accelerated force density method☆98Mar 4, 2026Updated 2 weeks ago
- ☆21Aug 21, 2025Updated 7 months ago
- ☆13Aug 28, 2023Updated 2 years ago
- A fast transfer matrix method written in jax for modelling optical multilayer thin films☆30Feb 11, 2026Updated last month
- ☆17Feb 10, 2023Updated 3 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆108Nov 12, 2023Updated 2 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- Numerical quadrature with JAX☆78Updated this week
- Examples for photonic design and simulation problems using various tools and methods.☆12Mar 21, 2023Updated 2 years ago
- Stencil computations in JAX☆71Nov 15, 2025Updated 4 months ago
- Conda + KLayout☆11Aug 21, 2022Updated 3 years ago
- Jaxwell is JAX + Maxwell☆27Oct 18, 2024Updated last year
- A collection of inverse design challenges☆69Oct 22, 2025Updated 4 months ago
- Additive manufacturing simulation with JAX.☆341Jul 11, 2025Updated 8 months ago
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆554Mar 9, 2026Updated last week
- A python code for 2d topology optimization using MMA optimizers in NLOPT☆12Feb 9, 2017Updated 9 years ago
- A suite of photonic inverse design challenge problems for topology optimization benchmarking☆119Jan 20, 2024Updated 2 years ago