Sergei Lebedev 51eb0d27c7 Fixed some type errors under pyright ...

These are mostly due to relience on submodule import side-effects, which
AFAIU are unchecked by both pytype and mypy.

2024-09-05 09:56:38 +01:00

..

__init__.py

[export] Move the export implementation to jax._src.export.

2024-06-09 08:59:50 -07:00

_export.py

Fixed some type errors under pyright

2024-09-05 09:56:38 +01:00

serialization_generated.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

serialization.fbs

[export] Rename some API entry points

2024-06-13 06:44:13 +02:00

serialization.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00

shape_poly_decision.py

[shape_poly] Move to jax._src in preparation for adding to AOT APIs.

2024-06-04 22:03:24 -07:00

shape_poly.py

Reverts 093b92be8ed7bd979486614325956e88cc474ff1

2024-07-23 04:32:56 -07:00