jax authors c0dacbf724 Merge pull request #23484 from justinjfu:pallas_prefetch_docs ...

PiperOrigin-RevId: 672538687

2024-09-09 07:33:57 -07:00

..

tpu

Merge pull request #23484 from justinjfu:pallas_prefetch_docs

2024-09-09 07:33:57 -07:00

CHANGELOG.md

Normalize "interpreter mode" to "interpret mode", and "InterpreterTest" to "InterpretTest"

2024-08-19 10:40:22 -07:00

design.md

Changed `pl.BlockSpec to accept block_shape before index_map`

2024-07-01 14:26:08 -07:00

grid_blockspec.md

[pallas] Enable check for GPU lowering that tensor sizes are power of 2

2024-08-05 02:34:21 -07:00

index.rst

[pallas] Add a warning message about experimental and incomplete status

2024-08-07 08:38:56 +03:00

quickstart.ipynb

Changed `pl.BlockSpec to accept block_shape before index_map`

2024-07-01 14:26:08 -07:00

quickstart.md

pre-commit: update hooks & pin using hashes

2024-08-27 15:23:13 -07:00