jax
3eaa7f90 - Remove `SafeNumDevices` now that all shardings are implemented in C++

Commit
205 days ago
Remove `SafeNumDevices` now that all shardings are implemented in C++ PiperOrigin-RevId: 785645705
Author
Parents
Loading