mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-25 01:06:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/experimental/_private_mm
History
jax authors a6b8384aed Merge pull request #26564 from gspschmid:gschmid/mini_mpmd 
PiperOrigin-RevId: 730912043
2025-02-25 09:15:40 -08:00
..
examples
Add jax.experimental._mini_mpmd
2025-02-25 17:02:29 +00:00
__init__.py
Add jax.experimental._mini_mpmd
2025-02-25 17:02:29 +00:00
BUILD
Merge pull request #26564 from gspschmid:gschmid/mini_mpmd
2025-02-25 09:15:40 -08:00
mini_dime.py
Add jax.experimental._mini_mpmd
2025-02-25 17:02:29 +00:00
mm.py
Add jax.experimental._mini_mpmd
2025-02-25 17:02:29 +00:00
profile_utils.py
Add jax.experimental._mini_mpmd
2025-02-25 17:02:29 +00:00