PyTorch
API: —
Strategy: Custom / Partial
JAX (Core)
API: jax.numpy.ix_
Strategy: Direct Mapping
NumPy
API: numpy.ix_
Strategy: Direct Mapping
TensorFlow
API: tf.experimental.numpy.ix_
Strategy: Direct Mapping
Flax NNX
API: jax.numpy.ix_
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.ix_
Strategy: Direct Mapping