Closed copybara-service[bot] closed 2 months ago
Move calibration instantiation in fwd pass to outside of custom_vjp to allow a stateful bwd pass calibrators.
Move calibration instantiation in fwd pass to outside of custom_vjp to allow a stateful bwd pass calibrators.