flax
dacf0036 - Replaces references to jax.numpy.DeviceArray with jax.Array.

Commit

2 years ago

Replaces references to jax.numpy.DeviceArray with jax.Array. PiperOrigin-RevId: 517192725

Author

hawkinsp

Committer

a-googler

Parents