Get our free extension to see links to code for papers anywhere online!
Free add-on: code for papers everywhere!
Free add-on: See code for papers anywhere!
Add to Chrome
Add to Firefox
Add to Edge
CatalyzeX Icon
Search Icon
Code for
Small-scale proxies for large-scale Transformer training instabilities
Github Icon
google/jax
Explore Code
Download Icon
Github Icon
google/orbax
Explore Code
Download Icon
Github Icon
google/grain
Explore Code
Download Icon
Github Icon
google/flax
Explore Code
Download Icon
Add to Chrome
Add to Firefox
Add to Edge