jax

480b4992 - jax/BUILD: switch to pytype_strict_library everywhere

jax/BUILD: switch to pytype_strict_library everywhere This enforces strict dependency checks for better maintainability going forward, and also does pytype checks for the included files. This required some minor refactoring of imports to conform to the downstream module specs used for strict dependency checks. PiperOrigin-RevId: 780776770