diff --git a/blackjax/types.py b/blackjax/types.py
index a8b9cc59b..d7d769f0a 100644
--- a/blackjax/types.py
+++ b/blackjax/types.py
@@ -12,7 +12,6 @@
 # limitations under the License.
 from typing import Any, Iterable, Mapping, Union
 
-import jax
 from jax import Array
 from jax.typing import ArrayLike