Refactor nvtx annotations in cudf & dask-cudf (#10396)

This PR consolidates all `nvtx.annotate` calls using common decorators `cudf_nvtx_annotate` & `dask_cudf_nvtx_annotate` that makes it easier to maintain and annotate APIs.

Authors:
  - GALI PREM SAGAR (https://github.com/galipremsagar)

Approvers:
  - Vyas Ramasubramani (https://github.com/vyasr)
  - Benjamin Zaitlen (https://github.com/quasiben)

URL: #10396

python/cudf/cudf/core/groupby/groupby.py

@@ -7,7 +7,6 @@
 from functools import cached_property
 
 import numpy as np
-from nvtx import annotate
 
 import cudf
 from cudf._lib import groupby as libgroupby
@@ -18,7 +17,7 @@
 from cudf.core.column.column import arange, as_column
 from cudf.core.mixins import Reducible, Scannable
 from cudf.core.multiindex import MultiIndex
-from cudf.utils.utils import GetAttrGetItemMixin
+from cudf.utils.utils import GetAttrGetItemMixin, _cudf_nvtx_annotate
 
 
 # The three functions below return the quantiles [25%, 50%, 75%]
@@ -207,7 +206,7 @@ def cumcount(self):
     def _groupby(self):
         return libgroupby.GroupBy(self.grouping.keys, dropna=self._dropna)
 
-    @annotate("GROUPBY_AGG", domain="cudf_python")
+    @_cudf_nvtx_annotate
     def agg(self, func):
         """
         Apply aggregation(s) to the groups.