jax authors 4f70471310 Fix error in pallas tutorial ...

PiperOrigin-RevId: 737727935

2025-03-17 13:19:12 -07: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

Fix error in pallas tutorial

2025-03-17 13:19:12 -07:00

sparse.md

Fix error in pallas tutorial

2025-03-17 13:19:12 -07:00