Bump JaxKern

thomaspinder · thomaspinder · commit 6f69f226b050 · 2023-01-09T08:59:42.000Z
diff --git a/setup.py b/setup.py
@@ -57,7 +57,7 @@ def get_versions():
         "jaxlib>=0.4.1",
         "optax",
         "jaxutils>=0.0.6",
-        "jaxkern",
+        "jaxkern>=0.0.4",
         "distrax>=0.1.2",
         "tqdm>=4.0.0",
         "ml-collections==0.1.0",
