Support "diagonal" primitives with no/slow JVP batch rule#164

Open

jpbrodrick89 wants to merge 40 commits intopatrick-kidger:mainfrom

jpbrodrick89:jpb/diagfromjac

Commits on May 14, 2025

Merge pull request #1 from patrick-kidger/main
jpbrodrick89
authored

Commits on Jun 19, 2025

unravel instead of map
jpbrodrick89
committed

Commits on Oct 24, 2025

Commits on Dec 13, 2025

Decrease default value to prevent overflow in 32-bit.
Johanna Haffner
committed
use maximum value of runtime dtype
Johanna Haffner
committed
limit condition numbers for sensitive, iterative solvers
Johanna Haffner
committed
cond_cutoff -> 900 for GMRES, LSMR
Johanna Haffner
committed
decrease condition number cutoff for LSMR
Johanna Haffner
committed
implement safeguard for int overflow in 32 bit
Johanna Haffner
committed
add extra safeguard for complex dtypes
Johanna Haffner
committed
fix typo
Johanna Haffner
committed
reset condition number check
Johanna Haffner
committed
simplify dtype conversion: reuse function from our misc module
Johanna Haffner
committed
Merge pull request #176 from johannahaffner/lsmr-init-overflow
johannahaffner
authored

Commits on Dec 22, 2025

Moved bool states to Static.
patrick-kidger
committed

Commits on Jan 28, 2026

Merge branch 'main' into jpb/diagfromjac
jpbrodrick89
authored

Commits on Jan 30, 2026

add efficient Matrix mv path, and Jacobian/Function materialise path
jpbrodrick89
committed