PyTorch
API: —
Strategy: Custom / Partial
JAX (Core)
API: jax.numpy.ndenumerate
Strategy: Direct Mapping
NumPy
API: numpy.ndenumerate
Strategy: Direct Mapping
Keras
API: —
Strategy: Custom / Partial
TensorFlow
API: —
Strategy: Custom / Partial
Apple MLX
API: —
Strategy: Custom / Partial