jax.lax.cummax# jax.lax.cummax(operand, axis=0, reverse=False)[source]# Computes a cumulative maximum along axis. Parameters: operand (Array) axis (int) reverse (bool) Return type: Array