richard
/
learn_jax
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
learn_jax/parallel/.gitignore

3 lines
26 B
Plaintext
Raw Normal View History

Feat: fsdp demo Refactor: pulling dataloader code into dataload.py
2024-09-15 22:41:00 +09:00
__pycache__
Feat: t5_jax_simple_parallel implements a working example of fsdp
2024-09-20 23:42:51 +09:00
gpt-neo-125m/