jax-fenics
jax-fenics copied to clipboard
Update registration of custom derivatives in JAX
See https://github.com/IvanYashchuk/jax-fenics-adjoint/issues/8
See https://github.com/IvanYashchuk/jax-fenics-adjoint/issues/8