aquakin.forward_adjoint

aquakin.forward_adjoint()

Return the diffrax adjoint that supports forward-mode autodiff.

A thin, dependency-free alias for diffrax.DirectAdjoint() so a user script that needs forward-mode AD through a reactor solve (e.g. the reactor inside a dgsm(ad_mode="forward") fn) can write adjoint=aquakin.forward_adjoint() without importing diffrax or knowing that the default RecursiveCheckpointAdjoint registers a custom_vjp that rejects forward mode.

Return type:

AbstractAdjoint