Added numba backend for Nonzero

Author: Abhinav-Khot

pytensor/tensor/basic.py

@@ -13,7 +13,6 @@
 from typing import TYPE_CHECKING, Union
 from typing import cast as type_cast
 
-import numba as nb
 import numpy as np
 from numpy.exceptions import AxisError
 
@@ -973,7 +972,6 @@ def make_node(self, a):
         output = [TensorType(dtype="int64", shape=(None,))() for i in range(a.ndim)]
         return Apply(self, [a], output)
 
-    @nb.njit
     def perform(self, node, inp, out_):
         a = inp[0]