flax
9534e266 - Avoid deprecated pattern of constructing a dtype from an array.

Commit

42 days ago

Avoid deprecated pattern of constructing a dtype from an array. `np.dtype(array)` will error in a future JAX version. PiperOrigin-RevId: 897252469

References

Author

Jake VanderPlas

Committer

a-googler

Parents