diff --git a/qwix/__init__.py b/qwix/__init__.py
index 3960a75..cb2fbd8 100644
--- a/qwix/__init__.py
+++ b/qwix/__init__.py
@@ -16,6 +16,7 @@
 
 # pylint: disable=g-multiple-import, g-importing-member
 
+from qwix._src.core import qarray, ragged_dot
 from qwix._src.flax_util import QuantStat
 from qwix._src.model import quantize_model
 from qwix._src.providers.lora import LoraProvider, LoraRule, apply_lora_to_model