Method argsort

ot/backend.py:1648–1649

(self, a, axis=-1)

        return jnp.sort(a, axis)

    def argsort(self, a, axis=-1):
        return jnp.argsort(a, axis)

    def searchsorted(self, a, v, side="left"):
        if a.ndim == 1: