mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,267 Commits 500 Branches 263 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
jax authors
4f70471310 Fix error in pallas tutorial 
PiperOrigin-RevId: 737727935
 2025-03-17 13:19:12 -07:00
Justin Fu
2b2d7cda98 [Pallas] Update TPU documentation 2024-12-06 10:59:11 -08:00
Dongseong Hwang
e4091a6752 Fix another errata in block-sparse kernel tutorial. 
PiperOrigin-RevId: 677952796
 2024-09-23 15:04:29 -07:00
Dongseong Hwang
91f16419bb Fix errata in block-sparse kernel tutorial. 
Correct M//blk_M to N//blk_N. It was ok because both values happen to be same.
In addition, grid order is (num_blocks, j) as 'num_blocks' replaces 'i'.

PiperOrigin-RevId: 677817478
 2024-09-23 09:07:28 -07:00
Justin Fu
4bce4f6452 [Pallas] Add block-sparse kernel tutorial 2024-09-19 12:23:03 -07:00