diff --git a/examples/irreps_array.py b/examples/irreps_array.py
index bc686f4..4b3d9e0 100644
--- a/examples/irreps_array.py
+++ b/examples/irreps_array.py
@@ -1,4 +1,3 @@
-import jax
 import jax.numpy as jnp
 
 import e3nn_jax as e3nn