mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 21:06:06 +00:00

This is more consistent, since currently we use a mix of both styles. It may also help pytype yield more accurate types. PiperOrigin-RevId: 412057514