rocm_jax/mypy.ini

[mypy]

[mypy-absl.*]
ignore_missing_imports = True
[mypy-jaxlib.*]
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
Add mypy type checking (#2430) * Add type annotations to make mypy pass. * Add mypy to .travis.yml. 2020-03-18 17:06:05 -04:00			`[mypy]`

			`[mypy-absl.*]`
			`ignore_missing_imports = True`
			`[mypy-jaxlib.*]`
			`ignore_missing_imports = True`
			`[mypy-numpy.*]`
			`ignore_missing_imports = True`
			`[mypy-opt_einsum.*]`
			`ignore_missing_imports = True`
			`[mypy-scipy.*]`
			`ignore_missing_imports = True`
add type annotations to core.py tracing machinery also add .copy() method to core.trace_state global trace state 2020-03-28 14:55:58 -07:00			`[mypy-jax.interpreters.autospmd]`
			`ignore_errors = True`
omnistaging, under a flag and disabled by default (#3370) This change, when enabled, stages out all primitive calls in the dynamic scope of a jitted, pmapped, or control flow function, rather than only staging out based on data dependence. One improvement is that jitted functions can consume less memory, by avoiding instantiating large constants at trace time, and cause less memory fragmentation as well. It also simplifies several internals. See https://github.com/google/jax/pull/3370 fo more information. 2020-07-30 12:59:36 -07:00			`[mypy-jax.lax.lax_parallel]`
			`ignore_errors = True`