jax
19bb5639 - Merge pull request #1240 from google/issue1239

Commit

6 years ago

Merge pull request #1240 from google/issue1239 ensure DeviceArray.aval is ShapedArray

References

Author

mattjj

Parents