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

7 Commits

Author SHA1 Message Date
Praveen Batra
3a3190fbce Fix typo in Pallas TPU matmul doc. I think the logical layout of the input array is non-transposed, rather than transposed? 
PiperOrigin-RevId: 686151692
 2024-10-15 10:23:39 -07:00
Justin Fu
51a666fb8c [Pallas] Update Pallas docs with new figures and TPUCompilerParams 2024-09-06 14:30:29 -07:00
Sergei Lebedev
7157839853 Fixed pl.BlockSpec argument ordering in the Pallas TPU matmul tutorial 2024-07-22 12:28:40 +01:00
Sharad Vikram
b8c4dde429 Copy improvements 
Co-authored-by: Jacob Austin<jaaustin@google.com>
 2024-07-18 22:59:00 -07:00
Sharad Vikram
c504fa6a4a Minor matmul docs improvement 2024-07-18 15:21:18 -07:00
Sharad Vikram
ff62d5e229 Address changes 2024-07-16 19:24:56 -07:00
Sharad Vikram
39ec5dacb4 [Pallas TPU] Add matrix multiplication tutorial 2024-07-16 18:12:19 -07:00