PyTorch
API: torch.randperm
Strategy: Direct Mapping
JAX (Core)
API: jax.random.permutation
Strategy: Plugin (inject_prng)
NumPy
API: np.random.permutation
Strategy: Direct Mapping
Keras
API: keras.numpy.random.permutation
Strategy: Macro 'keras.ops.convert_to_tensor(np.random.permutation({n}))'
TensorFlow
API: tf.random.shuffle
Strategy: Macro 'tf.random.shuffle(tf.range({n}))'
Apple MLX
API: mlx.core.random.permutation
Strategy: Macro 'mlx.core.random.permutation(mlx.core.arange({n}))'
Flax NNX
API: jax.random.permutation
Strategy: Plugin (inject_prng)