diff --git a/machine_learning/automatic_differentiation.py b/machine_learning/automatic_differentiation.py
index 5c2708247c21..93e77e761ef9 100644
--- a/machine_learning/automatic_differentiation.py
+++ b/machine_learning/automatic_differentiation.py
@@ -12,10 +12,9 @@
 from collections import defaultdict
 from enum import Enum
 from types import TracebackType
-from typing import Any
+from typing import Any, Self
 
 import numpy as np
-from typing_extensions import Self  # noqa: UP035
 
 
 class OpType(Enum):