This website requires JavaScript.
Explore
Help
Sign In
richard
/
learn_jax
Watch
1
Star
0
Fork
You've already forked learn_jax
0
Code
Issues
Pull Requests
Packages
Projects
Releases
Wiki
Activity
7e1f45f466
learn_jax
/
parallel
/
.gitignore
3 lines
26 B
Plaintext
Raw
Normal View
History
Unescape
Escape
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/