(self, a, axis=None, keepdims=False)
| 1595 | return jnp.max(a, axis, keepdims=keepdims) |
| 1596 | |
| 1597 | def min(self, a, axis=None, keepdims=False): |
| 1598 | return jnp.min(a, axis, keepdims=keepdims) |
| 1599 | |
| 1600 | def maximum(self, a, b): |
| 1601 | return jnp.maximum(a, b) |