Dan Foreman-Mackey 67f24df740 Activate FFI implementation of symmetric Eigendecomposition. ...

These kernels support shape polymorphism in all dimensions and no GPU is required during lowering. The kernels have been included in jaxlib for more than 3 weeks so we don't need to include any forward compatibility checks.

PiperOrigin-RevId: 682415506

2024-10-04 12:38:26 -07:00

..

__init__.py

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

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

_export.py

Activate FFI implementation of symmetric Eigendecomposition.

2024-10-04 12:38:26 -07: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_polyO] Improve handling of equality shape constraints

2024-09-06 13:55:38 +03:00

shape_poly.py

[shape_polyO] Improve handling of equality shape constraints

2024-09-06 13:55:38 +03:00