EnzymeAD/Enzyme-JAX

Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.