jax
75e4279e - Set `jax_varying_axes_in_types` to True by default.

Commit

296 days ago

Set `jax_varying_axes_in_types` to True by default. PiperOrigin-RevId: 745739477

References

#27880 - Set `jax_varying_axes_in_types` to True by default.

Author

yashk2810

yashk2810

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading