richard
/
learn_jax
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
6 Commits 1 Branch 0 Tags 176 KiB
Commit Graph

2 Commits

Author SHA1 Message Date
Richard Wong 429e1742ab Feat: flax pjit example 2024-09-16 12:19:07 +09:00
Richard Wong ad5cf7735f Feat: fsdp demo 
Refactor: pulling dataloader code into dataload.py
 2024-09-15 22:41:00 +09:00