numpy.nan_to_num has not been included in the array API spec so far.

However, it seems widely adopted by array libraries:

• torch.nan_to_num
• cupy.nan_to_num
• dask.array.nan_to_num
• jax.numpy.nan_to_num
• mlx.core.nan_to_num

At the time of writing, ndonnx does not implement this function.

I haven't checked in details, but from a cursory glance at the docs, libraries that implement this utility do seem quite consistent already.