jax
9fc47e51 - use DeviceTuples in optimizer states again

Commit

7 years ago

use DeviceTuples in optimizer states again

References

#684 - fix some DeviceTuples unpacking issues, re-enable in optimizers

Author

mattjj

mattjj

Parents

Loading