LexSort ======= Perform an indirect stable sort using a sequence of keys. **Abstract Signature:** ``LexSort(keys: Tensor, axis: int = -1)`` .. raw:: html

JAX (Core)

API: jax.numpy.lexsort
Strategy: Direct Mapping

NumPy

API: numpy.lexsort
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.lexsort
Strategy: Direct Mapping

PaxML / Praxis

API: jax.numpy.lexsort
Strategy: Direct Mapping