Rebecca Chen 6bd0c72f55 Silence some pytype errors. ...

PiperOrigin-RevId: 404122713

2021-10-18 18:47:24 -07:00

..

__init__.py

[JAX] Move traceback_util.py into jax._src.

2020-11-04 09:02:59 -08:00

ad.py

Cleanup internal representation of XLA translation rules.

2021-10-16 07:53:24 -07:00

batching.py

Switch internal users of jax.util.partial to use functools.partial.

2021-09-13 21:09:58 -04:00

invertible_ad.py

Move contents of jax.api_util to jax._src.api_util and add a forwarding shim.

2021-09-08 09:00:56 -07:00

masking.py

[masking] Remove references to masking.Poly from the lax.py and lax_numpy.py

2021-04-08 17:45:14 +03:00

partial_eval.py

Change jax.core.DropVar to be a non-singleton.

2021-10-18 15:02:54 -07:00

pxla.py

Remove xla_bridge.make_computation_builder().

2021-10-18 13:20:34 -04:00

sharded_jit.py

Remove xla_bridge.make_computation_builder().

2021-10-18 13:20:34 -04:00

xla.py

Silence some pytype errors.

2021-10-18 18:47:24 -07:00