jax
eb4ac933 - Merge pull request #680 from google/disable-devicetuples

Commit

6 years ago

Merge pull request #680 from google/disable-devicetuples temporarily avoid DeviceTuples in optimizer states

References

Author

mattjj

Parents