Sort

Sorts the elements of the input tensor.

Abstract Signature:

Sort(input: Tensor, dim: int = -1, descending: bool = False, stable: bool = False)

PyTorch

API: torch.sort
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.sort
Strategy: Plugin (sort_tuple_wrapper)

NumPy

API: numpy.sort
Strategy: Plugin (sort_tuple_wrapper)

Keras

API: keras.ops.sort
Strategy: Plugin (sort_tuple_wrapper)

TensorFlow

API: tf.sort
Strategy: Plugin (sort_tuple_wrapper)

Apple MLX

API: mlx.core.sort
Strategy: Plugin (sort_tuple_wrapper)

Flax NNX

API: jax.numpy.sort
Strategy: Plugin (sort_tuple_wrapper)

PaxML / Praxis

API: jax.numpy.sort
Strategy: Plugin (sort_tuple_wrapper)