PyTorch
API: torch.nanargmin
Strategy: Direct Mapping
JAX (Core)
API: jax.numpy.nanargmin
Strategy: Direct Mapping
NumPy
API: numpy.nanargmin
Strategy: Direct Mapping
Keras
API: —
Strategy: Plugin (numpy_fallback)
TensorFlow
API: —
Strategy: Plugin (numpy_fallback)
Apple MLX
API: —
Strategy: Plugin (numpy_fallback)
Flax NNX
API: jax.numpy.nanargmin
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.nanargmin
Strategy: Direct Mapping