search

kinnala / scikit-fem

Simple finite element assemblers
https://scikit-fem.readthedocs.io
BSD 3-Clause "New" or "Revised" License
470 stars 76 forks source link

JAX version of autodiff #1092

Closed kinnala closed 5 months ago

kinnala commented 6 months ago

Use JAX in experimental.autodiff.