Jake VanderPlas 3cecbf34f2 Remove core.concrete_aval and replace with abstractify

2024-12-17 18:18:25 -08:00

..

__init__.py

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

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

_export.py

[export] Expand exporting to work with AbstractMesh.

2024-12-16 10:30:46 +02:00

serialization_generated.py

Add float8_e4m3 and float8_e3m4 types support

2024-11-08 18:58:31 +00:00

serialization.fbs

Add float8_e4m3 and float8_e3m4 types support

2024-11-08 18:58:31 +00:00

serialization.py

Add float8_e4m3 and float8_e3m4 types support

2024-11-08 18:58:31 +00:00

shape_poly_decision.py

[shape_poly] Improve reasoning for >= in presence of == constraints.

2024-12-11 10:51:49 +01:00

shape_poly.py

Remove core.concrete_aval and replace with abstractify

2024-12-17 18:18:25 -08:00