qwix ragged dot in moe.py

PiperOrigin-RevId: 811884862

Author: Qwix Developers

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