Michael Hudgins d4d1518c3d Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax ...

PiperOrigin-RevId: 676843138

2024-09-20 07:52:33 -07:00

..

details.rst

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

distributed.ipynb

[Pallas] Update Pallas docs with new figures and TPUCompilerParams

2024-09-06 14:30:29 -07:00

distributed.md

Merge pull request #23484 from justinjfu:pallas_prefetch_docs

2024-09-09 07:33:57 -07:00

index.rst

[Pallas] Add block-sparse kernel tutorial

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

matmul.ipynb

[Pallas] Update Pallas docs with new figures and TPUCompilerParams

2024-09-06 14:30:29 -07:00

matmul.md

Merge pull request #23484 from justinjfu:pallas_prefetch_docs

2024-09-09 07:33:57 -07:00

pipelining.ipynb

[Pallas] Update Pallas docs with new figures and TPUCompilerParams

2024-09-06 14:30:29 -07:00

pipelining.md

Merge pull request #23484 from justinjfu:pallas_prefetch_docs

2024-09-09 07:33:57 -07:00

sparse.ipynb

[Pallas] Add block-sparse kernel tutorial

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

sparse.md

[Pallas] Add block-sparse kernel tutorial

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