bugfix init imports

Author: astanziola

jaxdf/__init__.py

@@ -1,9 +1,9 @@
-from core import *
-import discretization
-import geometry
-import ode
-import operators
-import primitives
-import spectral
-import utils
-import version
+from jaxdf.core import *
+from jaxdf import discretization
+from jaxdf import geometry
+from jaxdf import ode
+from jaxdf import operators
+from jaxdf import primitives
+from jaxdf import spectral
+from jaxdf import utils
+from jaxdf import version