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

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

..

details.rst

[pallas] Document and test valid block shapes.

2024-07-17 05:17:20 -07:00

index.rst

[Pallas TPU] Add matrix multiplication tutorial

2024-07-16 18:12:19 -07:00

matmul.ipynb

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

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

matmul.md

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

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

pipelining.ipynb

Changed `pl.BlockSpec to accept block_shape before index_map`

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

pipelining.md

Changed `pl.BlockSpec to accept block_shape before index_map`

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