JAX NumPy Operators Status ========================== # Overview SPU recommends users to use JAX as frontend language to write logics. We found most of users would utilize **jax.numpy** modules in their programs. We have conducted tests with some *selected* Operators for reference. Just keep in mind, if you couldn't find a **jax.numpy** operator in this list, it doesn't mean it's not supported. We just haven't test it, e.g. jax.numpy.sort. And we don't test other **JAX** modules at this moment. Please contact us if - You need to confirm the status of another **jax.numpy** operator not listed here. - You find a **jax.numpy** is not working even it is marked as **PASS**. e.g. The precision is bad. # Tested Operators List ## abs JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.abs.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## add JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.add.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## angle JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.angle.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.atan2 ## angle JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.angle.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.atan2 ## arccos JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.arccos.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.atan2 ## arccosh JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.arccosh.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note nan ## arcsin JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.arcsin.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.atan2 ## arcsinh JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.arcsinh.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## arctan JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.arctan.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.atan2 ## arctan2 JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.arctan2.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.atan2 ## arctanh JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.arctanh.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note [-1, 1] nan ## argmax JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.argmax.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## argmin JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.argmin.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## array_equal JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.array_equal.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## array_equiv JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.array_equiv.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## atleast_1d JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.atleast_1d.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## atleast_2d JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.atleast_2d.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## atleast_3d JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.atleast_3d.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## bitwise_and JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.bitwise_and.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## bitwise_not JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.bitwise_not.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## bitwise_or JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.bitwise_or.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## bitwise_xor JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.bitwise_xor.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## cbrt JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.cbrt.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.cbrt ## ceil JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ceil.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## conj JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.conj.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## conjugate JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.conjugate.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## copysign JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.copysign.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note shift ## cos JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.cos.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.cosine ## cosh JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.cosh.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## deg2rad JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.deg2rad.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## divide JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.divide.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## divmod JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.divmod.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## ediff1d JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ediff1d.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int32 ## equal JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.equal.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## exp JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.exp.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## exp2 JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.exp2.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## expm1 JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.expm1.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## fabs JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.fabs.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## fix JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.fix.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## float_power JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.float_power.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## floor JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.floor.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## floor_divide JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.floor_divide.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## fmax JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.fmax.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## fmin JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.fmin.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## fmod JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.fmod.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## gcd JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.gcd.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note secret while ## greater JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.greater.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## greater_equal JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.greater_equal.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## heaviside JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.heaviside.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## hypot JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.hypot.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## i0 JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.i0.html ### Status Result is incorrect. Please check *Note* for details. ### Note accuracy ## imag JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.imag.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## invert JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.invert.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## isclose JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isclose.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## iscomplex JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.iscomplex.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## isfinite JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isfinite.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## isinf JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isinf.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## isnan JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isnan.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## isneginf JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isneginf.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## isposinf JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isposinf.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## isreal JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isreal.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## isrealobj JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.isrealobj.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## kron JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.kron.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## lcm JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.lcm.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note secret while ## ldexp JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ldexp.html ### Status Result is incorrect. Please check *Note* for details. ### Note IEEE-754 ## left_shift JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.left_shift.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## less JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.less.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## less_equal JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.less_equal.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## log JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.log.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## log10 JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.log10.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## log1p JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.log1p.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## log1p JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.log1p.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## log2 JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.log2.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## logaddexp JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.logaddexp.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## logaddexp2 JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.logaddexp2.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## logical_and JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.logical_and.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## logical_not JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.logical_not.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## logical_or JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.logical_or.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## logical_xor JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.logical_xor.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## maximum JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.maximum.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## mean JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.mean.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## minimum JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.minimum.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## mod JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.mod.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## modf JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.modf.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## multiply JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.multiply.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## nanargmax JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.nanargmax.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## nanargmin JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.nanargmin.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## nanmean JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.nanmean.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## nanprod JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.nanprod.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## nansum JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.nansum.html ### Status Not supported by design. We couldn't fix this in near future. Please check *Note* for details. ## negative JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.negative.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## nextafter JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.nextafter.html ### Status Result is incorrect. Please check *Note* for details. ### Note IEEE-754 ## not_equal JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.not_equal.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## outer JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.outer.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## polyval JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.polyval.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## positive JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.positive.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## power JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.power.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## prod JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.prod.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## rad2deg JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.rad2deg.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## ravel JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ravel.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## real JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.real.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## reciprocal JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.reciprocal.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 ## remainder JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.remainder.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## right_shift JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.right_shift.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## rint JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.rint.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.round_nearest_even ## sign JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.sign.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## signbit JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.signbit.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## sin JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.sin.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.sine ## sinc JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.sinc.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.sine ## sinh JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.sinh.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.cosine ## sqrt JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.sqrt.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## square JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.square.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## subtract JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.subtract.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## sum JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.sum.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - int16 - int32 - uint16 - uint32 ## tan JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.tan.html ### Status Not supported by compiler or runtime. But we could implement on demand in future. Please check *Note* for details. ### Note stablehlo.sine ## tanh JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.tanh.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## transpose JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.transpose.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - bool_ - float32 - int16 - int32 - uint16 - uint32 ## true_divide JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.true_divide.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## trunc JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.trunc.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32 - int16 - int32 - uint16 - uint32 ## unwrap JAX NumPy Document link: https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.unwrap.html ### Status **PASS** Please check *Supported Dtypes* as well. ### Supported Dtypes - float32