George Necula e6066e52a2 [pallas] Stop exporting jax.experimental.pallas.pallas ...

This was giving access to too many internal APIs.

PiperOrigin-RevId: 655887765

2024-07-25 03:03:58 -07:00

..

tpu

Fixed pl.BlockSpec argument ordering in the Pallas TPU matmul tutorial

2024-07-22 12:28:40 +01:00

CHANGELOG.md

[pallas] Stop exporting jax.experimental.pallas.pallas

2024-07-25 03:03:58 -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

Reverts 0d058ce86f04a44a51abba1261768fb46edf69d9

2024-07-25 01:50:36 -07:00

index.rst

[pallas] Added a CHANGELOG for Pallas

2024-07-12 00:05:31 +03:00

quickstart.ipynb

Changed `pl.BlockSpec to accept block_shape before index_map`

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

quickstart.md

Changed `pl.BlockSpec to accept block_shape before index_map`

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