Remove unused return wrapper in annotate_function that creates a self reference cycle loop in python.

PiperOrigin-RevId: 745099538

Author: Google-ML-Automation

jax/_src/profiler.py

@@ -332,7 +332,6 @@ def annotate_function(func: Callable, name: str | None = None,
   def wrapper(*args, **kwargs):
     with TraceAnnotation(name, **decorator_kwargs):
       return func(*args, **kwargs)
-    return wrapper
   return wrapper