Contents:
Checks if an array is a matrix
A (jax.numpy.ndarray) – A JAX array
True if the array is a matrix, False otherwise.
bool