jax authors c7ca35fe32 Merge pull request #26345 from wenscarl:scaled_matmul ...

PiperOrigin-RevId: 731865430

2025-02-27 14:24:48 -08:00

..

__init__.py

Cleanup: fix unused imports & mark exported names

2024-10-16 17:42:41 -07:00

fused_attention_stablehlo.py

disable head dim 256 on bw now

2025-02-05 22:07:48 +00:00

fusion.py

jax.core: deprecate a number of APIs

2024-12-10 11:11:32 -08:00

scaled_matmul_stablehlo.py

Add apache header.

2025-02-26 15:26:56 -06:00