Remove unused imports

jacobusmmsmit · jacobusmmsmit · commit fc785a54a8df · 2022-08-17T14:03:29.000+01:00
diff --git a/diffrax/adjoint.py b/diffrax/adjoint.py
@@ -2,7 +2,6 @@
 from typing import Any, Dict
 
 import equinox as eqx
-import jax
 import jax.lax as lax
 import jax.numpy as jnp
 import jax.tree_util as jtu
diff --git a/test/helpers.py b/test/helpers.py
@@ -5,7 +5,6 @@
 
 import diffrax
 import equinox as eqx
-import jax
 import jax.numpy as jnp
 import jax.random as jrandom
 import jax.tree_util as jtu
