Start util cleanup

Author: willgebhardt

ngclearn/__init__.py

@@ -28,11 +28,11 @@
 
 import ngcsimlib
 
-from ngcsimlib import Component, MethodProcess, JointProcess
+from ngclearn.utils import JointProcess, MethodProcess
 from ngcsimlib.context import Context, ContextObjectTypes
 from ngcsimlib.compartment import Compartment
 
-from ngclearn.utils.jaxProcess import JaxProcess
+# from ngclearn.utils.jaxProcess import JaxProcess
 from ngcsimlib import logger
 
 # if not Path(argv[0]).name == "sphinx-build" or Path(argv[0]).name == "build.py":

ngclearn/utils/__init__.py

@@ -1,10 +1,3 @@
-from .model_utils import tensorstats
-from .jaxProcess import JaxProcess
-## forward imports from core ngc-learn utility sub-packages
-from . import viz
-from . import io_utils
-from . import metric_utils
-from . import model_utils
-from . import patch_utils
-from . import weight_distribution
-from . import surrogate_fx
+from .distribution_generator import DistributionGenerator
+from .JaxProcessesMixin import JaxJointProcess as JointProcess, JaxMethodProcess as MethodProcess
+