jax authors 2c165bffc9 [pallas:triton] Lift dot_general restriction on minimal tile size for a. ...

PiperOrigin-RevId: 725605869

2025-02-11 06:27:16 -08:00

..

mosaic

[better_errors] Continue adding debug info to Jaxprs (step 6)

2025-02-11 11:28:58 +01:00

mosaic_gpu

Merge pull request #26423 from gnecula:debug_info_jaxpr_7

2025-02-10 12:58:26 -08:00

triton

[pallas:triton] Lift dot_general restriction on minimal tile size for a.

2025-02-11 06:27:16 -08:00

__init__.py

[Pallas] Upstream pallas to JAX

2023-08-01 16:43:13 -07:00

BUILD

[Pallas] Refactor Pallas HLO interpret mode to a standalone file.

2025-01-27 17:52:27 -08:00

core.py

Merge pull request #26423 from gnecula:debug_info_jaxpr_7

2025-02-10 12:58:26 -08:00

cost_estimate.py

[better_errors] Continue adding debug info to Jaxprs (step 7)

2025-02-09 18:14:33 +02:00

helpers.py

[Pallas TPU] Add helpers for writing collectives

2025-02-04 15:39:10 -08:00

hlo_interpreter.py

[better_errors] Continue adding debug info to Jaxprs (step 3)

2025-02-06 16:26:49 +02:00

pallas_call.py

[better_errors] Continue adding debug info to Jaxprs (step 6)

2025-02-11 11:28:58 +01:00

primitives.py

[better_errors] Continue adding debug info to Jaxprs (step 7)

2025-02-09 18:14:33 +02:00

utils.py

[Pallas TPU] Add helpers for writing collectives

2025-02-04 15:39:10 -08:00