NanMax

Return the maximum of an array or maximum along an axis, ignoring any NaNs.

Abstract Signature:

NanMax(a: Array, axis: int = None, keepdims: bool = False)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Plugin (torch_nanmax_shim)

JAX (Core)

API: jax.numpy.nanmax

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.nanmax

Strategy: Direct Mapping

Official Docs ↗

Keras

API: —

Strategy: Custom / Partial

TensorFlow

API: tf.experimental.numpy.nanmax

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: —

Strategy: Custom / Partial

Flax NNX

API: jax.numpy.nanmax

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: jax.numpy.nanmax

Strategy: Direct Mapping

Official Docs ↗