jax.extend.core.array_types

jax.extend.core.array_types = {<class 'ml_dtypes.bfloat16'>, <class 'ml_dtypes.float4_e2m1fn'>, <class 'ml_dtypes.float8_e3m4'>, <class 'ml_dtypes.float8_e4m3'>, <class 'ml_dtypes.float8_e4m3b11fnuz'>, <class 'ml_dtypes.float8_e4m3fn'>, <class 'ml_dtypes.float8_e4m3fnuz'>, <class 'ml_dtypes.float8_e5m2'>, <class 'ml_dtypes.float8_e5m2fnuz'>, <class 'ml_dtypes.float8_e8m0fnu'>, <class 'ml_dtypes.int2'>, <class 'ml_dtypes.int4'>, <class 'ml_dtypes.uint2'>, <class 'ml_dtypes.uint4'>, <class 'numpy.bool'>, <class 'numpy.complex128'>, <class 'numpy.complex64'>, <class 'numpy.float16'>, <class 'numpy.float32'>, <class 'numpy.float64'>, <class 'numpy.int16'>, <class 'numpy.int32'>, <class 'numpy.int64'>, <class 'numpy.int8'>, <class 'numpy.longlong'>, <class 'numpy.ndarray'>, <class 'numpy.uint16'>, <class 'numpy.uint32'>, <class 'numpy.uint64'>, <class 'numpy.uint8'>}

set() -> new empty set object set(iterable) -> new set object

Build an unordered collection of unique elements.