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

This PR is a follow up to #18881. The changes were generated by adding from __future__ import annotations to the files which did not already have them and running pyupgrade --py39-plus --keep-percent-format {jax,tests,jaxlib,examples,benchmarks}/**/*.py
This sub-directory contains third-party code for which Google does not have copyright. Each sub-directory should correspond to a third-party library and must contain the appropriate LICENSE file. See instructions.