Remove spurious jax.core import

patrick-kidger · patrick-kidger · commit c5cb137c802b · 2024-12-23T23:02:04.000+01:00
diff --git a/lineax/_misc.py b/lineax/_misc.py
@@ -16,7 +16,6 @@
 
 import equinox as eqx
 import jax
-import jax.core
 import jax.numpy as jnp
 import jax.tree_util as jtu
 from jaxtyping import Array, ArrayLike, Bool, PyTree  # pyright:ignore
