[JAX] Suppress type errors found by pytype after correcting definition of jax.typing.ArrayLike.#338

copybara-service[bot] merged 1 commit intomaingoogle/learned_optimization:mainfrom

test_884868811google/learned_optimization:test_884868811Copy head branch name to clipboard