Justin Fu e05afefc97 [Pallas] Pallas documentation cleanup

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

..

details.rst

[Pallas] [Docs] Replace full urls with label-based cross references

2024-10-03 14:35:51 -07: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 another errata in block-sparse kernel tutorial.

2024-09-23 15:04:29 -07:00

sparse.md

Fix another errata in block-sparse kernel tutorial.

2024-09-23 15:04:29 -07:00