Peter Hawkins 474dcd409d Remove code to support jaxlib < v0.6. ...

New minimum jaxlib_extension_version is 330.

PiperOrigin-RevId: 748853497

2025-04-17 16:44:41 -07:00

..

mlir

Remove the try/except for Shardy imports.

2025-04-01 11:33:44 -07:00

__init__.py

Remove code to support jaxlib < v0.6.

2025-04-17 16:44:41 -07:00

BUILD

Split weakref_lru_cache into its own extension.

2025-04-08 13:34:47 -07:00

mosaic_gpu.py

[mosaic_gpu] Fixed mosaic_gpu-serde pass registration

2025-01-24 06:35:54 -08:00

triton.py

[pallas:triton] The lowering now uses PTX instead of Triton IR

2025-02-03 13:21:40 -08:00