jax
aeccdff4 - Clean up a few TODOs in pjit after FlatTree change

Commit

13 days ago

Clean up a few TODOs in pjit after FlatTree change PiperOrigin-RevId: 857342349

References

#34452 - Clean up a few TODOs in pjit after FlatTree change

Author

yashk2810

yashk2810

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading