jax
709086fa - [typing] add type annotation for shard_map

Commit

73 days ago

[typing] add type annotation for shard_map fixes #35101 PiperOrigin-RevId: 875165447

References

Author

Jake VanderPlas

Committer

Google-ML-Automation

Parents