flax
a8bc3673
- Replaces pjit with jit in spmd.py
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
Replaces pjit with jit in spmd.py PiperOrigin-RevId: 574373680
References
#3421 - Replaces pjit with jit in spmd.py
Author
marcvanzee
Committer
a-googler
Parents
31eedf16
Loading