Remove jax import

jwilles · jwilles · commit 2cabb3a31314 · 2024-08-09T12:03:09.000-04:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "vbll"
-version = "0.1.3"
+version = "0.2.1"
 description = ""
 authors = ["John Willes <johnwilles@gmail.com>"]
 readme = "README.md"
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 
 setup(
     name="vbll",
-    version="0.0.1",
+    version="0.2.1",
     packages=find_packages(),
     install_requires=["torch"],
 )
diff --git a/vbll/__init__.py b/vbll/__init__.py
@@ -1,2 +1 @@
 from .layers import *
-from . import jax
