Implemented NumbaExecutionEngine

doc/source/whatsnew/v3.0.0.rst

@@ -30,6 +30,7 @@ Other enhancements
 ^^^^^^^^^^^^^^^^^^
 - :class:`pandas.api.typing.FrozenList` is available for typing the outputs of :attr:`MultiIndex.names`, :attr:`MultiIndex.codes` and :attr:`MultiIndex.levels` (:issue:`58237`)
 - :class:`pandas.api.typing.SASReader` is available for typing the output of :func:`read_sas` (:issue:`55689`)
+- :meth:`DataFrame.apply` accepts Numba as an engine by passing the JIT decorator directly, e.g. ``df.apply(func, engine=numba.jit)`` (:issue:`61458`)
 - Added :meth:`.Styler.to_typst` to write Styler objects to file, buffer or string in Typst format (:issue:`57617`)
 - Added missing :meth:`pandas.Series.info` to API reference (:issue:`60926`)
 - :class:`pandas.api.typing.NoDefault` is available for typing ``no_default``

pandas/core/_numba/executor.py

@@ -18,14 +18,14 @@
 
 
 @functools.cache
-def generate_apply_looper(func, nopython=True, nogil=True, parallel=False):
+def generate_apply_looper(func, decorator: Callable):
     if TYPE_CHECKING:
         import numba
     else:
         numba = import_optional_dependency("numba")
     nb_compat_func = jit_user_function(func)
 
-    @numba.jit(nopython=nopython, nogil=nogil, parallel=parallel)
+    @decorator
     def nb_looper(values, axis, *args):
         # Operate on the first row/col in order to get
         # the output shape