Move calibration instantiation in fwd pass to outside of custom_vjp to allow a stateful bwd pass calibrators.

google / aqt

Apache License 2.0

247 stars 25 forks source link

Closed copybara-service[bot] closed 2 months ago

copybara-service[bot] commented 2 months ago

Move calibration instantiation in fwd pass to outside of custom_vjp to allow a stateful bwd pass calibrators.