Contents:
Returns the maximum of the off diagonal elements
A (jax.numpy.ndarray) – A real 2D matrix
The largest off-diagonal element in A
(float)