NanMaxΒΆ

Return the maximum of the array elements along a given axis, ignoring NaNs.

Abstract Signature:

NanMax(a: Tensor, axis: int | Sequence[int] | None, keepdims: bool = False, initial, where: Tensor | None)

PyTorch

API: torch.nanmax
Strategy: Direct Mapping

JAX (Core)

API: jax.numpy.nanmax
Strategy: Direct Mapping

NumPy

API: numpy.nanmax
Strategy: Direct Mapping