jax
4afee754 - Remove USE_NEW_TPU_CALLBACK_LOWERING and bump libtpu version.

Commit

45 days ago

Remove USE_NEW_TPU_CALLBACK_LOWERING and bump libtpu version. With libtpu 0.0.35 we no longer need this. PiperOrigin-RevId: 869376086

Author

hawkinsp

Committer

Google-ML-Automation

Parents