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