FEAT: JAX Conversion for the given SortOp

[ferrine]

Description

NotImplementedError: No JAX conversion for the given Op: SortOp{quicksort, None}

[HarshvirSandhu]

I would like to work on this

[twiecki]

I would like to work on this

Great - you can just open a PR or let us know if you have any questions.

[HarshvirSandhu]

@twiecki
Should the implementation and tests be added with the basic tensor functionalities(here) and tests (here)?

[twiecki]

@twiecki Should the implementation and tests be added with the basic tensor functionalities(here) and tests (here)?

Yes I believe that's where it should go.