flax
dcf5517e
- Update Flax pjit guide notebook
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
3 years ago
Update Flax pjit guide notebook
References
#2730 - Guide for Flax partitioning (auto SPMD) API
Author
8bitmp3
Parents
023511b6
Loading
