Revision 634f58c7d5bae8aad68a881280385f58ca2d0239 authored by Peter Hawkins on 03 May 2022, 01:56:50 UTC, committed by jax authors on 03 May 2022, 01:57:27 UTC
In particular, pjit/xmap work on CPU these days.

PiperOrigin-RevId: 446085110
1 parent b7293d5
Raw File
mypy.ini
[mypy]
show_error_codes = True
disable_error_code = attr-defined

[mypy-absl.*]
ignore_missing_imports = True
[mypy-colorama.*]
ignore_missing_imports = True
[mypy-numpy.*]
ignore_missing_imports = True
[mypy-opt_einsum.*]
ignore_missing_imports = True
[mypy-scipy.*]
ignore_missing_imports = True
[mypy-jax.interpreters.autospmd]
ignore_errors = True
[mypy-jax.lax.lax_parallel]
ignore_errors = True
[mypy-jax.experimental.jax2tf.tests.primitive_harness]
ignore_errors = True
[mypy-libtpu.*]
ignore_missing_imports = True
[mypy-jaxlib.mlir.*]
ignore_missing_imports = True
[mypy-iree.*]
ignore_missing_imports = True
back to top