IvanYashchuk / jax-fenics-adjoint

Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint

IvanYashchuk/jax-fenics-adjoint Issues

vmap for build_jax_fem_eval?
Updated 2 years ago
JIT is breaking because of numpy operations
Updated 3 years ago2
Update registration of custom derivatives in JAX
Closed 3 years ago
Update examples
Updated 3 years ago
Investigate NaN's in the output of jax.jvp
Closed 3 years ago2