* feat: support numba compiled sort and argsort functions
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* default to supported kind and add warning
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* feat: support axis
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* use syntax compatible with python 3.10
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* remove checks
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* use range instead of prange
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* add extra case to check Axis error is raised
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* simplify tests
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

---------
Signed-off-by: Victor Garcia Reolid <victor@seita.nl>

* Let numpy methods handle integer size problems in AdvancedSubtensor1

* Don't run `local_uint_constant_indices` in C/python backend

Indices are always cast to int64 by the underlying methods.

Also don't run in specialize, to reduce number of passes. Other rewrites may introduce temporar indexing operations (such as x.shape[i]) which always default to int64, and it's useless to optimize immediately.

Co-authored-by: Jesse Grabowski <jessegrabowski@gmail.com>
Co-authored-by: Rob Zinkov <zaxtax@users.noreply.github.com>

* Add numba overload for Nonzero

* added numba backend and testsfor Nonzero

* Added numba backend for Nonzero

* Modified the tests and the dispatch for efficiency

Also removes redundant tests

Graph was being broken by Scalar/Tensor conversions that prevented fusion

This will always require a roll at the end, for a minimal gain