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
10
Commits
1
Branch
0
Tags
173
KiB
parallel
Commit Graph
3 Commits
Author
SHA1
Message
Date
Richard Wong
7e1f45f466
Chore: removing old files and some experiments
2024-10-06 23:53:57 +09:00
Richard Wong
a817fe16cc
Feat: increased learning rate for effective large batch size learning
2024-09-22 22:28:41 +09:00
Richard Wong
aca80720c8
Feat: t5_jax_simple_parallel implements a working example of fsdp
2024-09-20 23:42:51 +09:00