From 3a5ce95d095f67f1e6933c228a50ef405979066b Mon Sep 17 00:00:00 2001 From: Qwix Developers Date: Fri, 26 Sep 2025 11:07:07 -0700 Subject: [PATCH] qwix ragged dot in moe.py PiperOrigin-RevId: 811884862 --- qwix/__init__.py | 1 + 1 file changed, 1 insertion(+) 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