Efficient forward- and reverse-mode sparse Jacobians using Jax
☆69Dec 1, 2025Updated 2 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 week
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 4 months ago
- measure minimum solid/void lengthscales in binary image☆19Dec 8, 2025Updated 2 months ago
- 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
- testbed problems for photonics optimization☆43Jun 22, 2024Updated last year
- Fourier Modal Method with Jax (official maintained repo)☆16Oct 21, 2025Updated 4 months ago
- Vector Finite Difference Maxwell Modesolver☆16Jun 8, 2024Updated last year
- Real-time design of architectural structures with differentiable mechanics and neural networks [ICLR 2025]☆23Jun 13, 2025Updated 8 months ago
- A GPU-accelerated, differentiable, Maxwell FDTD solver☆71Updated this week
- ☆35Jul 19, 2023Updated 2 years ago
- ☆45Oct 3, 2022Updated 3 years ago
- geometric primitives for piecewise functions on grids☆23Jun 28, 2024Updated last year
- Dataset of solutions to inverse design challenges☆14Oct 23, 2025Updated 4 months ago
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Feb 8, 2025Updated last year
- Easily mesh collections of 2D and 3D polygons with complex extrusions using GMSH☆29Feb 2, 2026Updated 3 weeks ago
- Artificial nanofabrication of integrated photonic circuits using deep learning☆71Feb 4, 2026Updated 3 weeks ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆50Sep 14, 2024Updated last year
- 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
- SParse lINear solvers in JAX☆26Updated this week
- Efficient Differentiable n-d PDE Solvers in JAX.☆145Feb 20, 2026Updated last week
- A differentiable finite element analysis solver for structural optimization based on JAX☆40Dec 22, 2025Updated 2 months ago
- ☆21Aug 21, 2025Updated 6 months ago
- ☆13Aug 28, 2023Updated 2 years ago
- Examples for photonic design and simulation problems using various tools and methods.☆12Mar 21, 2023Updated 2 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆516Updated this week
- Conda + KLayout☆11Aug 21, 2022Updated 3 years ago
- A fast transfer matrix method written in jax for modelling optical multilayer thin films☆29Feb 11, 2026Updated 2 weeks ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Nov 12, 2023Updated 2 years ago
- Python Implementation of Aperiodic Fourier Modal Method for solving Maxwell equations.☆23Feb 27, 2024Updated 2 years ago
- finite difference based eigensolving for Maxwell's equations and photonic structures☆14Oct 28, 2021Updated 4 years ago
- ☆17Feb 10, 2023Updated 3 years ago
- Auto-differentiable and hardware-accelerated force density method☆95Feb 15, 2026Updated last week
- ☆34Updated this week
- ☆47Jun 28, 2025Updated 8 months ago
- Numerical quadrature with JAX☆77Feb 16, 2026Updated last week
- GPU-accelerated electromagnetic FDTD simulations for inverse design / gradient-based optimization (parameter, shape, topology) of nanopho…☆24Updated this week
- Stencil computations in JAX☆71Nov 15, 2025Updated 3 months ago
- FEM mode solver for photonic waveguides☆156Feb 21, 2026Updated last week