missing import

mfinzi · Feb 26, 2021 · 4cb4826 · 4cb4826
1 parent 452c661
commit 4cb4826
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/emlp/solver/linear_operators.py b/emlp/solver/linear_operators.py
@@ -2,6 +2,7 @@
 import jax.numpy as jnp
 import numpy as np
 import jax.jit as jit
+import jax
 from functools import reduce
 
 class Lazy(LinearOperator):