brainpy · chaoming0625 · Nov 23, 2024 · Nov 23, 2024
diff --git a/brainpy/_src/math/__init__.py b/brainpy/_src/math/__init__.py
@@ -47,7 +47,6 @@
 from . import random, linalg, fft
 
 # operators
-from .op_register import *
 from .pre_syn_post import *
 from . import surrogate, event, sparse, jitconn
 

diff --git a/brainpy/_src/math/jitconn/matvec.py b/brainpy/_src/math/jitconn/matvec.py
@@ -4,14 +4,12 @@
 
 import jax
 import numpy as np
-from jax import numpy as jnp
-
+from brainpy._src.dependency_check import import_braintaichi, raise_braintaichi_not_found
 from brainpy._src.math import defaults
 from brainpy._src.math.interoperability import as_jax
 from brainpy._src.math.ndarray import Array
-from brainpy._src.math.op_register import XLACustomOp
 from brainpy.errors import PackageMissingError
-from brainpy._src.dependency_check import import_braintaichi, raise_braintaichi_not_found
+from jax import numpy as jnp
 
 bti = import_braintaichi(error_if_not_found=False)
 

diff --git a/brainpy/_src/math/op_register/__init__.py b/brainpy/_src/math/op_register/__init__.py
diff --git a/brainpy/_src/math/op_register/ad_support.py b/brainpy/_src/math/op_register/ad_support.py
diff --git a/brainpy/_src/math/op_register/base.py b/brainpy/_src/math/op_register/base.py