jax
0ccf7cf6 - Merge pull request #684 from google/device-tuples

Commit

7 years ago

Merge pull request #684 from google/device-tuples fix some DeviceTuples unpacking issues, re-enable in optimizers

References

Author

mattjj

Parents