https://github.com/google/jax
Raw File
Tip revision: 5d7f639769b038672750a1d83ae7c0bdd695aa9f authored by Qiao Zhang on 17 March 2022, 21:47:02 UTC
Add small and big matmul to api_benchmarks.
Tip revision: 5d7f639
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