Peter Hawkins 7f4ef63cd8 Run pyupgrade --py310-plus. ...

Also apply manual fixes to import sorting and unused imports.

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

..

__init__.py

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

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

_export.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04: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

Run pyupgrade --py310-plus.

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