jax authors 1c07ec6183 Merge pull request #25272 from justinjfu:pallas_tpu_docs_update ...

PiperOrigin-RevId: 704376603

2024-12-09 12:22:31 -08:00

..

details.rst

[Pallas] Update TPU documentation

2024-12-06 10:59:11 -08:00

distributed.ipynb

Update Pallas distributed tutorials with jax.make_mesh

2024-10-21 12:49:56 -07:00

distributed.md

Update Pallas distributed tutorials with jax.make_mesh

2024-10-21 12:49:56 -07:00

index.rst

[Pallas] Add block-sparse kernel tutorial

2024-09-19 12:23:03 -07:00

matmul.ipynb

Fix typo in Pallas TPU matmul doc. I think the logical layout of the input array is non-transposed, rather than transposed?

2024-10-15 10:23:39 -07:00

matmul.md

Fix typo in Pallas TPU matmul doc. I think the logical layout of the input array is non-transposed, rather than transposed?

2024-10-15 10:23:39 -07:00

pipelining.ipynb

[Pallas] Pallas documentation cleanup

2024-12-04 15:19:32 -08:00

pipelining.md

[Pallas] Pallas documentation cleanup

2024-12-04 15:19:32 -08:00

sparse.ipynb

[Pallas] Update TPU documentation

2024-12-06 10:59:11 -08:00

sparse.md

[Pallas] Update TPU documentation

2024-12-06 10:59:11 -08:00