richard
/
learn_jax
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
learn_jax/parallel
History
Richard Wong ad5cf7735f Feat: fsdp demo 
Refactor: pulling dataloader code into dataload.py
 		2024-09-15 22:41:00 +09:00
..
.gitignore Feat: fsdp demo 2024-09-15 22:41:00 +09:00
dataload.py Feat: fsdp demo 2024-09-15 22:41:00 +09:00
fully_sharded_data_parallelism.py Feat: fsdp demo 2024-09-15 22:41:00 +09:00
intro_to_distributed.py Feat: fsdp demo 2024-09-15 22:41:00 +09:00
single_gpu_optimizations.py Feat: fsdp demo 2024-09-15 22:41:00 +09:00
t5_jax_train_pjit.py Feat: fsdp demo 2024-09-15 22:41:00 +09:00