jax.numpy.nanargmax =================== .. currentmodule:: jax.numpy .. autofunction:: nanargmax