PyTorch
API: torch.swapaxes
Strategy: Direct Mapping
JAX (Core)
API: jax.numpy.swapaxes
Strategy: Direct Mapping
NumPy
API: numpy.swapaxes
Strategy: Direct Mapping
Keras
API: keras.ops.swapaxes
Strategy: Direct Mapping
TensorFlow
API: tf.experimental.numpy.swapaxes
Strategy: Macro 'tf.transpose({a}, perm=...)'
Apple MLX
API: mlx.core.swapaxes
Strategy: Direct Mapping
Flax NNX
API: jax.numpy.swapaxes
Strategy: Direct Mapping
PaxML / Praxis
API: jax.numpy.swapaxes
Strategy: Direct Mapping