Open copybara-service[bot] opened 2 years ago
[JAX] Avoid private implementation detail _ScalarMeta.
The closest public approximation to type(jnp.float32) is type[Any]. Nothing is ever actually an instance of one of these types, either (they build DeviceArrays if instantiated.)
type[Any]
DeviceArray
[JAX] Avoid private implementation detail _ScalarMeta.
The closest public approximation to type(jnp.float32) is
type[Any]. Nothing is ever actually an instance of one of these types, either (they buildDeviceArrays if instantiated.)