docs: def reduce_cuda(g, x, axes, dtype):
      """Reductions in CUDA use the thrust library for speed and have limited
         functionality."""
      if axes != 0:
          raise NotImplementedError("'axes' keyword is not implemented for CUDA")
  
      return g(x, dtype=dtype)