Closed erick-xanadu closed 1 month ago
All modified and coverable lines are covered by tests :white_check_mark:
Project coverage is 97.98%. Comparing base (
f1c846d
) to head (792b0f1
). Report is 1 commits behind head on main.
:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
Context: Functions that have been accelerated can be are compatible with JAX. This means that JAX can also auto-differentiate them. So, let's leverage JAX to auto-differentiate them.
Description of the Change:
Benefits: Better UI
Possible Drawbacks: Callback code is becoming harder to reason about with all the special cases, accelerate, pure_callback, with/without gradient.
Related GitHub Issues:
[sc-60775]