rocm_jax/pallas at 10e09af7a06c3881284a0c1696ed79ebfde1971f - rocm_jax - Gitea For EOELAB

mirrors/rocm_jax

mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00

History

Sharad Vikram 10e09af7a0 Address changes

2024-07-16 19:25:32 -07:00

..

Address changes

2024-07-16 19:25:32 -07:00

CHANGELOG.md

[pallas] Added a CHANGELOG for Pallas

2024-07-12 00:05:31 +03:00

design.md

Changed `pl.BlockSpec to accept block_shape before index_map`

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

grid_blockspec.md

[pallas] Document the indexing_mode=Unblocked()

2024-07-12 12:39:10 +03:00

index.rst

[pallas] Added a CHANGELOG for Pallas

2024-07-12 00:05:31 +03:00

quickstart.ipynb

Changed `pl.BlockSpec to accept block_shape before index_map`

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

quickstart.md

Changed `pl.BlockSpec to accept block_shape before index_map`

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