Array Operations

Learn about the common array-related operations of NumPy and JAX.

Array operations

Operations on JAX arrays are similar to operations with NumPy arrays. For example, we can use the max(), argmax(), and sum() functions the same as we do in NumPy.

Get hands-on with 1200+ tech skills courses.