Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change. #2787
Triggered via pull request
September 24, 2024 18:33
Status
Success
Total duration
2m 59s
Artifacts
–
Annotations
6 warnings