George Necula afcb62ea20 [export] Expand exporting to work with AbstractMesh. ...

This is a follow up from #25640 that enabled lowering with
AbstractMesh.

This required adding `num_devices` to `lowering.compiler_args`
because in presence of an AbstractMesh the device_assignment
is not accurate.

2024-12-16 10:30:46 +02: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

Merge pull request #25442 from jakevdp:raise-to-shaped

2024-12-12 10:43:17 -08:00