While jax.numpy will implicitly promote arguments to allow operations between mixed data types, jax.lax will not; instead, it supplies explicit promotion functions. The lowest layer of the API is XLA.

NumPy, the mathematical library for Python, has received its first major release since 2006 with version 2.0.0. This not only offers new functions, but also performance improvements for Intel and ...