parakeet.ts 1.0.0

package/dist/mlx/nn.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nn.js","sourceRoot":"","sources":["../../src/mlx/nn.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AAYzC,8EAA8E;AAC9E,UAAU;AACV,8EAA8E;AAE9E,SAAS,CAAC,CAAC,GAAG,IAAc;IAC1B,OAAO,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;AAC9C,CAAC;AAED,SAAS,SAAS,CAAC,OAAkB,EAAE,GAAW;IAChD,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,MAAM,IAAI,KAAK,CAAC,sBAAsB,GAAG,GAAG,CAAC,CAAC;IAC1D,OAAO,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,SAAS,YAAY,CAAC,OAAkB,EAAE,GAAW;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;IAC/B,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IACxB,OAAO,OAAO,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,8EAA8E;AAC9E,cAAc;AACd,8EAA8E;AAE9E,MAAM,OAAgB,MAAM;CAE3B;AAED,8EAA8E;AAC9E,SAAS;AACT,8EAA8E;AAE9E,MAAM,OAAO,MAAO,SAAQ,MAAM;IAIb;IAA2B;IAA4B;IAH1E,MAAM,CAAW,CAAC,YAAY;IAC9B,IAAI,GAAmB,IAAI,CAAC;IAE5B,YAAmB,UAAkB,EAAS,WAAmB,EAAS,UAAU,IAAI;QACtF,KAAK,EAAE,CAAC;QADS,eAAU,GAAV,UAAU,CAAQ;QAAS,gBAAW,GAAX,WAAW,CAAQ;QAAS,YAAO,GAAP,OAAO,CAAO;IAExF,CAAC;IAED,OAAO,CAAC,CAAU;QAChB,4BAA4B;QAC5B,MAAM,GAAG,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC;IAC9C,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,SAAS,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;CACF;AAED,8EAA8E;AAC9E,YAAY;AACZ,8EAA8E;AAE9E,MAAM,OAAO,SAAU,SAAQ,MAAM;IAGhB;IAA8B;IAFjD,MAAM,CAAW,CAAC,kCAAkC;IAEpD,YAAmB,aAAqB,EAAS,YAAoB;QACnE,KAAK,EAAE,CAAC;QADS,kBAAa,GAAb,aAAa,CAAQ;QAAS,iBAAY,GAAZ,YAAY,CAAQ;IAErE,CAAC;IAED,OAAO,CAAC,OAAgB;QACtB,gDAAgD;QAChD,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACtC,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,SAAS,CAAC,CAAC;IACvD,CAAC;CACF;AAED,8EAA8E;AAC9E,YAAY;AACZ,8EAA8E;AAE9E,MAAM,OAAO,SAAU,SAAQ,MAAM;IAIhB;IAAyB;IAH5C,MAAM,CAAW,CAAC,aAAa;IAC/B,IAAI,CAAW,CAAG,aAAa;IAE/B,YAAmB,QAAgB,EAAS,MAAM,IAAI;QACpD,KAAK,EAAE,CAAC;QADS,aAAQ,GAAR,QAAQ,CAAQ;QAAS,QAAG,GAAH,GAAG,CAAO;IAEtD,CAAC;IAED,OAAO,CAAC,CAAU;QAChB,qBAAqB;QACrB,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACtB,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAChC,MAAM,QAAQ,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QACnC,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QAClE,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9C,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;IACnD,CAAC;CACF;AAED,8EAA8E;AAC9E,iDAAiD;AACjD,8EAA8E;AAE9E,MAAM,OAAO,SAAU,SAAQ,MAAM;IAMhB;IAAyB;IAL5C,MAAM,CAAW,CAAQ,aAAa;IACtC,IAAI,CAAW,CAAU,aAAa;IACtC,WAAW,CAAW,CAAG,aAAa;IACtC,UAAU,CAAW,CAAI,aAAa;IAEtC,YAAmB,QAAgB,EAAS,MAAM,IAAI;QACpD,KAAK,EAAE,CAAC;QADS,aAAQ,GAAR,QAAQ,CAAQ;QAAS,QAAG,GAAH,GAAG,CAAO;IAEtD,CAAC;IAED,OAAO,CAAC,CAAU;QAChB,wDAAwD;QACxD,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACrF,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC9C,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,IAAI,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,eAAe,CAAC,CAAC;QAChE,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,cAAc,CAAC,CAAC;IAChE,CAAC;CACF;AAED,8EAA8E;AAC9E,0DAA0D;AAC1D,EAAE;AACF,6DAA6D;AAC7D,oCAAoC;AACpC,0CAA0C;AAC1C,8EAA8E;AAE9E,MAAM,OAAO,MAAO,SAAQ,MAAM;IAKvB;IACA;IACA;IACA;IACA;IACA;IACA;IAVT,MAAM,CAAW,CAAC,+BAA+B;IACjD,IAAI,GAAmB,IAAI,CAAC;IAE5B,YACS,UAAkB,EAClB,WAAmB,EACnB,UAAkB,EAClB,SAAS,CAAC,EACV,UAAU,CAAC,EACX,SAAS,CAAC,EACV,UAAU,IAAI;QAErB,KAAK,EAAE,CAAC;QARD,eAAU,GAAV,UAAU,CAAQ;QAClB,gBAAW,GAAX,WAAW,CAAQ;QACnB,eAAU,GAAV,UAAU,CAAQ;QAClB,WAAM,GAAN,MAAM,CAAI;QACV,YAAO,GAAP,OAAO,CAAI;QACX,WAAM,GAAN,MAAM,CAAI;QACV,YAAO,GAAP,OAAO,CAAO;IAGvB,CAAC;IAED,OAAO,CAAC,CAAU;QAChB,OAAO,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACnF,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,SAAS,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;CACF;AAED,gEAAgE;AAChE,MAAM,UAAU,MAAM,CACpB,CAAU,EACV,MAAe,EACf,IAAoB,EACpB,MAAc,EACd,OAAe,EACf,MAAc;IAEd,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;IACzB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAChC,MAAM,GAAG,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAE/B,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAChC,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAErC,6BAA6B;IAC7B,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;QAChB,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACpE,CAAC;IACD,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAEjE,IAAI,MAAM,KAAK,CAAC,EAAE,CAAC;QACjB,4DAA4D;QAC5D,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC;QACpD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,OAAO,CAAC,CAAC,GAAG,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QACD,MAAM,GAAG,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC;QAE/D,0DAA0D;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAEnC,+CAA+C;QAC/C,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;QAExE,gDAAgD;QAChD,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,GAAG,IAAI,CAAC,CAAC,CAAC;QAEzE,4EAA4E;QAC5E,6CAA6C;QAC7C,MAAM,EAAE,GAAG,MAAM;aACd,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;aACpC,OAAO,CAAC,CAAC,CAAC,UAAU,GAAG,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;QAExC,IAAI,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,yBAAyB;QACxD,IAAI,IAAI;YAAE,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9B,OAAO,GAAG,CAAC;IACb,CAAC;IAED,IAAI,MAAM,KAAK,IAAI,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QACtC,2EAA2E;QAC3E,MAAM,OAAO,GAAG,IAAI,UAAU,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC;QACpD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gBACpC,OAAO,CAAC,CAAC,GAAG,UAAU,GAAG,CAAC,CAAC,GAAG,CAAC,GAAG,MAAM,GAAG,CAAC,CAAC;YAC/C,CAAC;QACH,CAAC;QACD,MAAM,GAAG,GAAG,OAAO,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,MAAM,GAAG,UAAU,CAAC,CAAC,CAAC;QAE/D,2CAA2C;QAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QACnC,6CAA6C;QAC7C,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,CAAC;QAExE,yEAAyE;QACzE,wCAAwC;QACxC,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAO,qBAAqB;QAC/E,MAAM,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAO,qBAAqB;QAEhF,+CAA+C;QAC/C,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAY,oCAAoC;QACpF,IAAI,GAAG,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAK,wBAAwB;QACzE,IAAI,IAAI;YAAE,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAC9B,OAAO,GAAG,CAAC;IACb,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,kBAAkB,MAAM,wBAAwB,IAAI,WAAW,KAAK,GAAG,CAAC,CAAC;AAC3F,CAAC;AAED,8EAA8E;AAC9E,uCAAuC;AACvC,EAAE;AACF,6CAA6C;AAC7C,iDAAiD;AACjD,8EAA8E;AAE9E,MAAM,OAAO,MAAO,SAAQ,MAAM;IAKvB;IACA;IACA;IACA;IACA;IACA;IACA;IAVT,MAAM,CAAW;IACjB,IAAI,GAAmB,IAAI,CAAC;IAE5B,YACS,UAAkB,EAClB,WAAmB,EACnB,UAAkB,EAClB,SAAS,CAAC,EACV,UAAU,CAAC,EACX,SAAS,CAAC,EACV,UAAU,IAAI;QAErB,KAAK,EAAE,CAAC;QARD,eAAU,GAAV,UAAU,CAAQ;QAClB,gBAAW,GAAX,WAAW,CAAQ;QACnB,eAAU,GAAV,UAAU,CAAQ;QAClB,WAAM,GAAN,MAAM,CAAI;QACV,YAAO,GAAP,OAAO,CAAI;QACX,WAAM,GAAN,MAAM,CAAI;QACV,YAAO,GAAP,OAAO,CAAO;IAGvB,CAAC;IAED,OAAO,CAAC,CAAU;QAChB,OAAO,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACnF,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,SAAS,CAAC,CAAC;QACrD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,IAAI,GAAG,YAAY,CAAC,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;CACF;AAED,wFAAwF;AACxF,MAAM,UAAU,MAAM,CACpB,CAAU,EACV,MAAe,EACf,IAAoB,EACpB,MAAc,EACd,OAAe,EACf,MAAc;IAEd,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;IACzB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAChC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAE/B,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAChC,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAC7B,MAAM,EAAE,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAE7B,MAAM;IACN,IAAI,IAAI,GAAG,CAAC,CAAC;IACb,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;QAChB,IAAI,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;IACtF,CAAC;IACD,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAChD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,EAAE,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;IAEhD,uDAAuD;IACvD,MAAM,YAAY,GAAG,IAAI,GAAG,IAAI,GAAG,EAAE,GAAG,EAAE,CAAC;IAC3C,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,YAAY,CAAC,CAAC;IAC1C,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,YAAY,CAAC,CAAC;IAE1C,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;QACjC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,IAAI,EAAE,EAAE,EAAE,EAAE,CAAC;YACjC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;gBAC/B,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;oBAC/B,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,MAAM,GAAG,EAAE,CAAC;oBAC7B,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,MAAM,GAAG,EAAE,CAAC;oBAC7B,GAAG,EAAE,CAAC;gBACR,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,MAAM,KAAK,GAAG,IAAI,UAAU,CAAC,YAAY,CAAC,CAAC;IAC3C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,KAAK,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IACpC,CAAC;IAED,8CAA8C;IAC9C,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;IACpD,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IAC1E,2CAA2C;IAE3C,IAAI,GAAY,CAAC;IAEjB,IAAI,MAAM,GAAG,CAAC,EAAE,CAAC;QACf,uDAAuD;QACvD,2CAA2C;QAC3C,kFAAkF;QAClF,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,GAAG,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;QAC1E,MAAM,SAAS,GAAG,UAAU,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACrE,oDAAoD;QACpD,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QACvD,oEAAoE;QACpE,GAAG,GAAG,SAAS,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACvD,CAAC;SAAM,CAAC;QACN,kBAAkB;QAClB,4CAA4C;QAC5C,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,GAAG,IAAI,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;QACzE,gEAAgE;QAChE,0CAA0C;QAC1C,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC;QACvD,MAAM,EAAE,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QACnD,GAAG,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,4BAA4B;IACzD,CAAC;IAED,wCAAwC;IACxC,GAAG,GAAG,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IAE/C,IAAI,IAAI;QAAE,GAAG,GAAG,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC9B,OAAO,GAAG,CAAC;AACb,CAAC;AAED,8EAA8E;AAC9E,qBAAqB;AACrB,8EAA8E;AAE9E,MAAM,UAAU,OAAO,CAAC,CAAU;IAChC,oBAAoB;IACpB,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,CAAU;IAC7B,OAAO,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,CAAU;IAC7B,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,UAAU,IAAI,CAAC,CAAU;IAC7B,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;AAClB,CAAC;AAED,MAAM,UAAU,GAAG,CAAC,CAAU,EAAE,IAAY;IAC1C,6DAA6D;IAC7D,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;IACxB,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IAChC,MAAM,IAAI,GAAG,GAAG,GAAG,CAAC,CAAC;IAErB,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAa,CAAC;IAC7D,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACpE,MAAM,OAAO,GAAG,IAAI,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,CAAa,CAAC;IAC7D,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAEjC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IAC3E,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;IAC3E,OAAO,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,CAAU,EAAE,IAAY;IAC9C,MAAM,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACnD,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC9B,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;IAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACtD,OAAO,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;AAC1B,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,CAAU,EAAE,IAAY;IACjD,OAAO,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;AAC5B,CAAC;AAED,8EAA8E;AAC9E,oCAAoC;AACpC,EAAE;AACF,oCAAoC;AACpC,gEAAgE;AAChE,oCAAoC;AACpC,4BAA4B;AAC5B,8EAA8E;AAE9E,MAAM,OAAO,SAAU,SAAQ,MAAM;IAKhB;IAA0B;IAA2B;IAJxE,EAAE,CAAW,CAAC,oBAAoB;IAClC,EAAE,CAAW,CAAC,qBAAqB;IACnC,CAAC,GAAmB,IAAI,CAAC;IAEzB,YAAmB,SAAiB,EAAS,UAAkB,EAAS,UAAU,IAAI;QACpF,KAAK,EAAE,CAAC;QADS,cAAS,GAAT,SAAS,CAAQ;QAAS,eAAU,GAAV,UAAU,CAAQ;QAAS,YAAO,GAAP,OAAO,CAAO;IAEtF,CAAC;IAED,mFAAmF;IACnF,OAAO,CACL,CAAU,EACV,CAAiB,EACjB,CAAiB;QAEjB,yBAAyB;QACzB,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QACjC,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC;QAE1B,0BAA0B;QAC1B,IAAI,CAAC,KAAK,IAAI;YAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,CAAC,KAAK,IAAI;YAAE,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QAErD,oEAAoE;QACpE,sDAAsD;QACtD,MAAM,KAAK,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,GAAG,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QAC3D,mDAAmD;QACnD,IAAI,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,IAAI,IAAI,CAAC,CAAC;YAAE,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtC,4BAA4B;QAC5B,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QAExD,MAAM,IAAI,GAAc,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAc,EAAE,CAAC;QAE3B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAChC,8CAA8C;YAC9C,MAAM,EAAE,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAE3F,4CAA4C;YAC5C,MAAM,KAAK,GAAG,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAElE,MAAM,KAAK,GAAG,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,cAAc;YAE3C,0CAA0C;YAC1C,MAAM,IAAI,GAAG,CAAC,GAAW,EAAE,EAAE,CAC3B,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,GAAG,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;YAEtD,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,aAAa;YAC1C,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc;YAC3C,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAG,YAAY;YACzC,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc;YAE3C,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,WAAW;YAC3C,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAW,aAAa;YAE7C,mDAAmD;YACnD,CAAC,CAAC,IAAI,EAAE,CAAC;YACT,CAAC,CAAC,IAAI,EAAE,CAAC;YAET,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACb,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACf,CAAC;QAED,yBAAyB;QACzB,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QACtC,6CAA6C;QAC7C,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACrB,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,EAAE,GAAG,SAAS,CAAC,OAAO,EAAE,GAAG,MAAM,KAAK,CAAC,CAAC;QAC7C,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,IAAI,CAAC,CAAC,GAAG,YAAY,CAAC,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,IAAI,YAAY,CAAC,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,CAAC;QAC3F,CAAC;IACH,CAAC;CACF;AAED,MAAM,OAAO,IAAK,SAAQ,MAAM;IAC9B,MAAM,CAAc;IAEpB,YACE,SAAiB,EACjB,UAAkB,EAClB,SAAiB,EACjB,OAAO,GAAG,IAAI;QAEd,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;QACjB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;QACzF,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,OAAO,CACL,CAAU,EACV,KAAgC,IAAI;QAEpC,+CAA+C;QAC/C,IAAI,OAAO,GAAG,CAAC,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAErD,MAAM,MAAM,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC;QACzB,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAChC,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QAEpC,MAAM,SAAS,GAAc,EAAE,CAAC;QAChC,MAAM,SAAS,GAAc,EAAE,CAAC;QAEhC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAC/F,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YAE/F,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC5D,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACvC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAEzF,OAAO,GAAG,IAAI,CAAC,CAAC,uCAAuC;YACvD,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACtB,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxB,CAAC;QAED,4CAA4C;QAC5C,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;QACpE,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,wBAAwB;QAEpE,OAAO,CAAC,MAAM,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IACpC,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,MAAM,IAAI,CAAC,EAAE,CAAC,CAAC;QACxD,CAAC;IACH,CAAC;CACF"}

package/dist/mlx/rnnt.d.ts

@@ -0,0 +1,59 @@
+import { MxArray } from '@mlx-node/core';
+import { Module, WeightMap, Linear, Embedding, LSTM } from './nn.js';
+import { AlignedToken } from '../alignment.js';
+export interface PredictNetworkArgs {
+    predHidden: number;
+    predRnnLayers: number;
+    rnnHiddenSize?: number;
+}
+export interface JointNetworkArgs {
+    jointHidden: number;
+    activation: string;
+    encoderHidden: number;
+    predHidden: number;
+}
+export interface PredictArgs {
+    blankAsPad: boolean;
+    vocabSize: number;
+    prednet: PredictNetworkArgs;
+}
+export interface JointArgs {
+    numClasses: number;
+    vocabulary: string[];
+    jointnet: JointNetworkArgs;
+    numExtraOutputs?: number;
+}
+export declare class PredictNetwork extends Module {
+    embed: Embedding;
+    decRnn: LSTM;
+    readonly predHidden: number;
+    constructor(args: PredictArgs);
+    /**
+     * y: [batch, 1] token ids or null for blank start
+     * hc: LSTM hidden/cell state or null
+     * Returns [output: [batch, 1, predHidden], [h, c]]
+     */
+    forward(y: MxArray | null, hc: [MxArray, MxArray] | null): [MxArray, [MxArray, MxArray]];
+    loadWeights(weights: WeightMap, prefix: string): void;
+}
+export declare class JointNetwork extends Module {
+    pred: Linear;
+    enc: Linear;
+    activation: (x: MxArray) => MxArray;
+    jointOut: Linear;
+    readonly numClasses: number;
+    constructor(args: JointArgs);
+    forward(enc: MxArray, pred: MxArray): MxArray;
+    loadWeights(weights: WeightMap, prefix: string): void;
+}
+export interface DecoderState {
+    lastToken: number | null;
+    hiddenState: [MxArray, MxArray] | null;
+}
+/** Greedy TDT decoder. Returns decoded token list per batch element. */
+export declare function decodeTDTGreedy(features: MxArray, // [batch, seqLen, encoderDim]
+lengths: MxArray, // [batch]
+decoder: PredictNetwork, joint: JointNetwork, vocabulary: string[], durations: number[], maxSymbols: number | null, states: DecoderState[], timeRatio: number): [Array<AlignedToken[]>, DecoderState[]];
+/** Greedy RNNT decoder */
+export declare function decodeRNNTGreedy(features: MxArray, lengths: MxArray, decoder: PredictNetwork, joint: JointNetwork, vocabulary: string[], maxSymbols: number | null, states: DecoderState[], timeRatio: number): [Array<Array<import('../alignment.js').AlignedToken>>, DecoderState[]];
+//# sourceMappingURL=rnnt.d.ts.map

package/dist/mlx/rnnt.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rnnt.d.ts","sourceRoot":"","sources":["../../src/mlx/rnnt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAA0B,MAAM,SAAS,CAAC;AAC7F,OAAO,EAAoB,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAWjE,MAAM,WAAW,kBAAkB;IACjC,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,gBAAgB;IAC/B,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;IACnB,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,WAAW;IAC1B,UAAU,EAAE,OAAO,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,EAAE,kBAAkB,CAAC;CAC7B;AAED,MAAM,WAAW,SAAS;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,QAAQ,EAAE,gBAAgB,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAMD,qBAAa,cAAe,SAAQ,MAAM;IACxC,KAAK,EAAE,SAAS,CAAC;IACjB,MAAM,EAAE,IAAI,CAAC;IACb,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;gBAEhB,IAAI,EAAE,WAAW;IAU7B;;;;OAIG;IACH,OAAO,CACL,CAAC,EAAE,OAAO,GAAG,IAAI,EACjB,EAAE,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,IAAI,GAC5B,CAAC,OAAO,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAahC,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;CAItD;AAMD,qBAAa,YAAa,SAAQ,MAAM;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,UAAU,EAAE,CAAC,CAAC,EAAE,OAAO,KAAK,OAAO,CAAC;IACpC,QAAQ,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;gBAEhB,IAAI,EAAE,SAAS;IAmB3B,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,GAAG,OAAO;IAY7C,WAAW,CAAC,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI;CAOtD;AAMD,MAAM,WAAW,YAAY;IAC3B,SAAS,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,WAAW,EAAE,CAAC,OAAO,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CACxC;AAED,wEAAwE;AACxE,wBAAgB,eAAe,CAC7B,QAAQ,EAAE,OAAO,EAAM,8BAA8B;AACrD,OAAO,EAAE,OAAO,EAAO,UAAU;AACjC,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,YAAY,EACnB,UAAU,EAAE,MAAM,EAAE,EACpB,SAAS,EAAE,MAAM,EAAE,EACnB,UAAU,EAAE,MAAM,GAAG,IAAI,EACzB,MAAM,EAAE,YAAY,EAAE,EACtB,SAAS,EAAE,MAAM,GAChB,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,EAAE,YAAY,EAAE,CAAC,CAwGzC;AAED,0BAA0B;AAC1B,wBAAgB,gBAAgB,CAC9B,QAAQ,EAAE,OAAO,EACjB,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,cAAc,EACvB,KAAK,EAAE,YAAY,EACnB,UAAU,EAAE,MAAM,EAAE,EACpB,UAAU,EAAE,MAAM,GAAG,IAAI,EACzB,MAAM,EAAE,YAAY,EAAE,EACtB,SAAS,EAAE,MAAM,GAChB,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,iBAAiB,EAAE,YAAY,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CA0ExE"}

package/dist/mlx/rnnt.js

@@ -0,0 +1,233 @@
+import { MxArray } from '@mlx-node/core';
+import { Module, Linear, Embedding, LSTM, relu, sigmoid, softmax } from './nn.js';
+import { makeAlignedToken } from '../alignment.js';
+import { decode } from '../tokenizer.js';
+function s(...dims) {
+    return BigInt64Array.from(dims.map(BigInt));
+}
+// ---------------------------------------------------------------------------
+// Predict (decoder) network — embedding + LSTM
+// ---------------------------------------------------------------------------
+export class PredictNetwork extends Module {
+    embed;
+    decRnn;
+    predHidden;
+    constructor(args) {
+        super();
+        this.predHidden = args.prednet.predHidden;
+        const vocabSize = args.blankAsPad ? args.vocabSize + 1 : args.vocabSize;
+        const rnnHidden = args.prednet.rnnHiddenSize ?? args.prednet.predHidden;
+        this.embed = new Embedding(vocabSize, args.prednet.predHidden);
+        this.decRnn = new LSTM(args.prednet.predHidden, rnnHidden, args.prednet.predRnnLayers);
+    }
+    /**
+     * y: [batch, 1] token ids or null for blank start
+     * hc: LSTM hidden/cell state or null
+     * Returns [output: [batch, 1, predHidden], [h, c]]
+     */
+    forward(y, hc) {
+        let embeddedY;
+        if (y !== null) {
+            embeddedY = this.embed.forward(y); // [batch, 1, predHidden]
+        }
+        else {
+            const batch = hc ? Number(hc[0].shape()[1]) : 1;
+            embeddedY = MxArray.zeros(s(batch, 1, this.predHidden), null);
+        }
+        return this.decRnn.forward(embeddedY, hc);
+    }
+    loadWeights(weights, prefix) {
+        this.embed.loadWeights(weights, `${prefix}.prediction.embed`);
+        this.decRnn.loadWeights(weights, `${prefix}.prediction.dec_rnn.lstm`);
+    }
+}
+// ---------------------------------------------------------------------------
+// Joint network — combines encoder and decoder outputs
+// ---------------------------------------------------------------------------
+export class JointNetwork extends Module {
+    pred;
+    enc;
+    activation;
+    jointOut;
+    numClasses;
+    constructor(args) {
+        super();
+        const numExtraOutputs = args.numExtraOutputs ?? 0;
+        this.numClasses = args.numClasses + 1 + numExtraOutputs;
+        this.pred = new Linear(args.jointnet.predHidden, args.jointnet.jointHidden);
+        this.enc = new Linear(args.jointnet.encoderHidden, args.jointnet.jointHidden);
+        this.jointOut = new Linear(args.jointnet.jointHidden, this.numClasses);
+        const act = args.jointnet.activation.toLowerCase();
+        if (act === 'relu') {
+            this.activation = relu;
+        }
+        else if (act === 'sigmoid') {
+            this.activation = sigmoid;
+        }
+        else {
+            this.activation = (x) => x.tanh();
+        }
+    }
+    forward(enc, pred) {
+        // enc:  [batch, encSeq, 1, encoderHidden]
+        // pred: [batch, 1, predSeq, predHidden]
+        const encP = this.enc.forward(enc); // [batch, encSeq, 1, jointHidden]
+        const predP = this.pred.forward(pred); // [batch, 1, predSeq, jointHidden]
+        // Broadcast addition: [batch, encSeq, predSeq, jointHidden]
+        const x = encP.add(predP);
+        const activated = this.activation(x);
+        return this.jointOut.forward(activated);
+    }
+    loadWeights(weights, prefix) {
+        this.pred.loadWeights(weights, `${prefix}.pred`);
+        this.enc.loadWeights(weights, `${prefix}.enc`);
+        // joint_net is a list: [activation, Identity, Linear]
+        // The Linear is at index 2 in Python (0=activation, 1=Identity, 2=Linear)
+        this.jointOut.loadWeights(weights, `${prefix}.joint_net.2`);
+    }
+}
+/** Greedy TDT decoder. Returns decoded token list per batch element. */
+export function decodeTDTGreedy(features, // [batch, seqLen, encoderDim]
+lengths, // [batch]
+decoder, joint, vocabulary, durations, maxSymbols, states, timeRatio) {
+    const fShape = features.shape();
+    const B = Number(fShape[0]);
+    const results = [];
+    const newStates = [];
+    for (let b = 0; b < B; b++) {
+        const hypothesis = [];
+        const feature = features.slice(s(b, 0, 0), s(b + 1, Number(fShape[1]), Number(fShape[2])));
+        const length = Number(lengths.toInt32()[b]);
+        let step = 0;
+        let newSymbols = 0;
+        let { lastToken, hiddenState } = states[b];
+        while (step < length) {
+            const yInput = lastToken !== null
+                ? MxArray.fromInt32(new Int32Array([lastToken]), s(1, 1))
+                : null;
+            const [decOut, [newH, newC]] = decoder.forward(yInput, hiddenState);
+            // decOut: [1, 1, predHidden] — expand for joint: [1, 1, 1, predHidden]
+            const decExpand = decOut.expandDims(1);
+            // encoder feature at current step: [1, 1, encoderDim] → [1, 1, 1, encoderDim]
+            const encStep = feature.slice(s(0, step, 0), s(1, step + 1, Number(feature.shape()[2]))).expandDims(2);
+            const jointOut = joint.forward(encStep, decExpand);
+            // jointOut: [1, 1, 1, numClasses]
+            const vocabSize = vocabulary.length;
+            const numClasses = Number(jointOut.shape()[3]);
+            const numDurations = durations.length;
+            const tokenLogits = jointOut
+                .slice(s(0, 0, 0, 0), s(1, 1, 1, vocabSize + 1))
+                .squeeze(new Int32Array([0, 1, 2]));
+            const durationLogits = jointOut
+                .slice(s(0, 0, 0, vocabSize + 1), s(1, 1, 1, numClasses))
+                .squeeze(new Int32Array([0, 1, 2]));
+            const tokenProbs = softmax(tokenLogits, 0).toFloat32();
+            const durationProbs = softmax(durationLogits, 0).toFloat32();
+            // Find argmax for token and duration
+            let predToken = 0;
+            let maxTokenProb = tokenProbs[0];
+            for (let i = 1; i < tokenProbs.length; i++) {
+                if (tokenProbs[i] > maxTokenProb) {
+                    maxTokenProb = tokenProbs[i];
+                    predToken = i;
+                }
+            }
+            let decision = 0;
+            let maxDurProb = durationProbs[0];
+            for (let i = 1; i < durationProbs.length; i++) {
+                if (durationProbs[i] > maxDurProb) {
+                    maxDurProb = durationProbs[i];
+                    decision = i;
+                }
+            }
+            const isBlank = predToken === vocabSize;
+            if (!isBlank) {
+                // Confidence using entropy
+                let entropy = 0;
+                for (let i = 0; i < tokenProbs.length; i++) {
+                    const p = tokenProbs[i] + 1e-10;
+                    entropy -= p * Math.log(p);
+                }
+                const maxEntropy = Math.log(tokenProbs.length);
+                const confidence = 1.0 - entropy / maxEntropy;
+                hypothesis.push(makeAlignedToken(predToken, decode([predToken], vocabulary), step * timeRatio, durations[decision] * timeRatio, confidence));
+                lastToken = predToken;
+                hiddenState = [newH, newC];
+            }
+            step += durations[decision];
+            newSymbols += 1;
+            if (durations[decision] !== 0) {
+                newSymbols = 0;
+            }
+            else if (maxSymbols !== null && newSymbols >= maxSymbols) {
+                step += 1;
+                newSymbols = 0;
+            }
+        }
+        results.push(hypothesis);
+        newStates.push({ lastToken, hiddenState });
+    }
+    return [results, newStates];
+}
+/** Greedy RNNT decoder */
+export function decodeRNNTGreedy(features, lengths, decoder, joint, vocabulary, maxSymbols, states, timeRatio) {
+    const fShape = features.shape();
+    const B = Number(fShape[0]);
+    const results = [];
+    const newStates = [];
+    for (let b = 0; b < B; b++) {
+        const hypothesis = [];
+        const feature = features.slice(s(b, 0, 0), s(b + 1, Number(fShape[1]), Number(fShape[2])));
+        const length = Number(lengths.toInt32()[b]);
+        let step = 0;
+        let newSymbols = 0;
+        let { lastToken, hiddenState } = states[b];
+        while (step < length) {
+            const yInput = lastToken !== null
+                ? MxArray.fromInt32(new Int32Array([lastToken]), s(1, 1))
+                : null;
+            const [decOut, [newH, newC]] = decoder.forward(yInput, hiddenState);
+            const decExpand = decOut.expandDims(1);
+            const encStep = feature.slice(s(0, step, 0), s(1, step + 1, Number(feature.shape()[2]))).expandDims(2);
+            const jointOut = joint.forward(encStep, decExpand);
+            const vocabSize = vocabulary.length;
+            const tokenLogits = jointOut
+                .slice(s(0, 0, 0, 0), s(1, 1, 1, vocabSize + 1))
+                .squeeze(new Int32Array([0, 1, 2]));
+            const tokenProbs = softmax(tokenLogits, 0).toFloat32();
+            let predToken = 0;
+            let maxP = tokenProbs[0];
+            for (let i = 1; i < tokenProbs.length; i++) {
+                if (tokenProbs[i] > maxP) {
+                    maxP = tokenProbs[i];
+                    predToken = i;
+                }
+            }
+            const isBlank = predToken === vocabSize;
+            if (!isBlank) {
+                let entropy = 0;
+                for (let i = 0; i < tokenProbs.length; i++) {
+                    const p = tokenProbs[i] + 1e-10;
+                    entropy -= p * Math.log(p);
+                }
+                const confidence = 1.0 - entropy / Math.log(tokenProbs.length);
+                hypothesis.push(makeAlignedToken(predToken, decode([predToken], vocabulary), step * timeRatio, timeRatio, confidence));
+                lastToken = predToken;
+                hiddenState = [newH, newC];
+                newSymbols += 1;
+                if (maxSymbols !== null && newSymbols >= maxSymbols) {
+                    step += 1;
+                    newSymbols = 0;
+                }
+            }
+            else {
+                step += 1;
+                newSymbols = 0;
+            }
+        }
+        results.push(hypothesis);
+        newStates.push({ lastToken, hiddenState });
+    }
+    return [results, newStates];
+}
+//# sourceMappingURL=rnnt.js.map

package/dist/mlx/rnnt.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rnnt.js","sourceRoot":"","sources":["../../src/mlx/rnnt.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,gBAAgB,CAAC;AACzC,OAAO,EAAE,MAAM,EAAa,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAC7F,OAAO,EAAE,gBAAgB,EAAgB,MAAM,iBAAiB,CAAC;AACjE,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAEzC,SAAS,CAAC,CAAC,GAAG,IAAc;IAC1B,OAAO,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;AAC9C,CAAC;AAgCD,8EAA8E;AAC9E,+CAA+C;AAC/C,8EAA8E;AAE9E,MAAM,OAAO,cAAe,SAAQ,MAAM;IACxC,KAAK,CAAY;IACjB,MAAM,CAAO;IACJ,UAAU,CAAS;IAE5B,YAAY,IAAiB;QAC3B,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC;QACxE,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,IAAI,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC;QAExE,IAAI,CAAC,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC/D,IAAI,CAAC,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IACzF,CAAC;IAED;;;;OAIG;IACH,OAAO,CACL,CAAiB,EACjB,EAA6B;QAE7B,IAAI,SAAkB,CAAC;QAEvB,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;YACf,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,yBAAyB;QAC9D,CAAC;aAAM,CAAC;YACN,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAChD,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,EAAE,IAAI,CAAC,CAAC;QAChE,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,MAAM,mBAAmB,CAAC,CAAC;QAC9D,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,MAAM,0BAA0B,CAAC,CAAC;IACxE,CAAC;CACF;AAED,8EAA8E;AAC9E,uDAAuD;AACvD,8EAA8E;AAE9E,MAAM,OAAO,YAAa,SAAQ,MAAM;IACtC,IAAI,CAAS;IACb,GAAG,CAAS;IACZ,UAAU,CAA0B;IACpC,QAAQ,CAAS;IACR,UAAU,CAAS;IAE5B,YAAY,IAAe;QACzB,KAAK,EAAE,CAAC;QACR,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,IAAI,CAAC,CAAC;QAClD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,GAAG,CAAC,GAAG,eAAe,CAAC;QAExD,IAAI,CAAC,IAAI,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAC5E,IAAI,CAAC,GAAG,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,aAAa,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QAC9E,IAAI,CAAC,QAAQ,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,EAAE,IAAI,CAAC,UAAU,CAAC,CAAC;QAEvE,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,WAAW,EAAE,CAAC;QACnD,IAAI,GAAG,KAAK,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;QACzB,CAAC;aAAM,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YAC7B,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;QAC5B,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,UAAU,GAAG,CAAC,CAAU,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,OAAO,CAAC,GAAY,EAAE,IAAa;QACjC,0CAA0C;QAC1C,wCAAwC;QACxC,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAG,kCAAkC;QACxE,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,mCAAmC;QAE1E,4DAA4D;QAC5D,MAAM,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC1B,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;IAC1C,CAAC;IAED,WAAW,CAAC,OAAkB,EAAE,MAAc;QAC5C,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;QACjD,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,CAAC;QAC/C,sDAAsD;QACtD,0EAA0E;QAC1E,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,OAAO,EAAE,GAAG,MAAM,cAAc,CAAC,CAAC;IAC9D,CAAC;CACF;AAWD,wEAAwE;AACxE,MAAM,UAAU,eAAe,CAC7B,QAAiB,EAAM,8BAA8B;AACrD,OAAgB,EAAO,UAAU;AACjC,OAAuB,EACvB,KAAmB,EACnB,UAAoB,EACpB,SAAmB,EACnB,UAAyB,EACzB,MAAsB,EACtB,SAAiB;IAEjB,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;IAChC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAE5B,MAAM,OAAO,GAA0B,EAAE,CAAC;IAC1C,MAAM,SAAS,GAAmB,EAAE,CAAC;IAErC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3F,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5C,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QAE3C,OAAO,IAAI,GAAG,MAAM,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,SAAS,KAAK,IAAI;gBAC/B,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACzD,CAAC,CAAC,IAAI,CAAC;YAET,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACpE,uEAAuE;YACvE,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAEvC,8EAA8E;YAC9E,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAEvG,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACnD,kCAAkC;YAElC,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC;YACpC,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAC/C,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,CAAC;YAEtC,MAAM,WAAW,GAAG,QAAQ;iBACzB,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;iBAC/C,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEtC,MAAM,cAAc,GAAG,QAAQ;iBAC5B,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;iBACxD,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEtC,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACvD,MAAM,aAAa,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YAE7D,qCAAqC;YACrC,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,IAAI,YAAY,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YACjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3C,IAAI,UAAU,CAAC,CAAC,CAAC,GAAG,YAAY,EAAE,CAAC;oBACjC,YAAY,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;oBAC7B,SAAS,GAAG,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC;YAED,IAAI,QAAQ,GAAG,CAAC,CAAC;YACjB,IAAI,UAAU,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAClC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9C,IAAI,aAAa,CAAC,CAAC,CAAC,GAAG,UAAU,EAAE,CAAC;oBAClC,UAAU,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;oBAC9B,QAAQ,GAAG,CAAC,CAAC;gBACf,CAAC;YACH,CAAC;YAED,MAAM,OAAO,GAAG,SAAS,KAAK,SAAS,CAAC;YAExC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,2BAA2B;gBAC3B,IAAI,OAAO,GAAG,CAAC,CAAC;gBAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC3C,MAAM,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;oBAChC,OAAO,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC7B,CAAC;gBACD,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,UAAU,GAAG,GAAG,GAAG,OAAO,GAAG,UAAU,CAAC;gBAE9C,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAC9B,SAAS,EACT,MAAM,CAAC,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,EAC/B,IAAI,GAAG,SAAS,EAChB,SAAS,CAAC,QAAQ,CAAC,GAAG,SAAS,EAC/B,UAAU,CACX,CAAC,CAAC;gBACH,SAAS,GAAG,SAAS,CAAC;gBACtB,WAAW,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAC7B,CAAC;YAED,IAAI,IAAI,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC5B,UAAU,IAAI,CAAC,CAAC;YAEhB,IAAI,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC9B,UAAU,GAAG,CAAC,CAAC;YACjB,CAAC;iBAAM,IAAI,UAAU,KAAK,IAAI,IAAI,UAAU,IAAI,UAAU,EAAE,CAAC;gBAC3D,IAAI,IAAI,CAAC,CAAC;gBACV,UAAU,GAAG,CAAC,CAAC;YACjB,CAAC;QACH,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzB,SAAS,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;AAC9B,CAAC;AAED,0BAA0B;AAC1B,MAAM,UAAU,gBAAgB,CAC9B,QAAiB,EACjB,OAAgB,EAChB,OAAuB,EACvB,KAAmB,EACnB,UAAoB,EACpB,UAAyB,EACzB,MAAsB,EACtB,SAAiB;IAEjB,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;IAChC,MAAM,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IAE5B,MAAM,OAAO,GAA0B,EAAE,CAAC;IAC1C,MAAM,SAAS,GAAmB,EAAE,CAAC;IAErC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC3B,MAAM,UAAU,GAAmB,EAAE,CAAC;QACtC,MAAM,OAAO,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3F,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAE5C,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QAE3C,OAAO,IAAI,GAAG,MAAM,EAAE,CAAC;YACrB,MAAM,MAAM,GAAG,SAAS,KAAK,IAAI;gBAC/B,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACzD,CAAC,CAAC,IAAI,CAAC;YAET,MAAM,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;YACpE,MAAM,SAAS,GAAG,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACvC,MAAM,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,GAAG,CAAC,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAEvG,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAEnD,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC;YACpC,MAAM,WAAW,GAAG,QAAQ;iBACzB,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;iBAC/C,OAAO,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAEtC,MAAM,UAAU,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACvD,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,IAAI,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC3C,IAAI,UAAU,CAAC,CAAC,CAAC,GAAG,IAAI,EAAE,CAAC;oBAAC,IAAI,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC;oBAAC,SAAS,GAAG,CAAC,CAAC;gBAAC,CAAC;YACpE,CAAC;YAED,MAAM,OAAO,GAAG,SAAS,KAAK,SAAS,CAAC;YAExC,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,OAAO,GAAG,CAAC,CAAC;gBAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC3C,MAAM,CAAC,GAAG,UAAU,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC;oBAChC,OAAO,IAAI,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;gBAC7B,CAAC;gBACD,MAAM,UAAU,GAAG,GAAG,GAAG,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBAE/D,UAAU,CAAC,IAAI,CAAC,gBAAgB,CAC9B,SAAS,EACT,MAAM,CAAC,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,EAC/B,IAAI,GAAG,SAAS,EAChB,SAAS,EACT,UAAU,CACX,CAAC,CAAC;gBACH,SAAS,GAAG,SAAS,CAAC;gBACtB,WAAW,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC3B,UAAU,IAAI,CAAC,CAAC;gBAChB,IAAI,UAAU,KAAK,IAAI,IAAI,UAAU,IAAI,UAAU,EAAE,CAAC;oBACpD,IAAI,IAAI,CAAC,CAAC;oBACV,UAAU,GAAG,CAAC,CAAC;gBACjB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,IAAI,IAAI,CAAC,CAAC;gBACV,UAAU,GAAG,CAAC,CAAC;YACjB,CAAC;QACH,CAAC;QAED,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzB,SAAS,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAC;IAC7C,CAAC;IAED,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;AAC9B,CAAC"}

package/dist/mlx/server.d.ts

@@ -0,0 +1,34 @@
+/**
+ * parakeet.ts/mlx/server — Hono route factory for the streaming ASR endpoint.
+ *
+ * Usage:
+ *   import { Hono } from 'hono';
+ *   import { serve } from '@hono/node-server';
+ *   import { fromPretrained } from 'parakeet.ts/mlx';
+ *   import { createParakeetRoutes } from 'parakeet.ts/mlx/server';
+ *
+ *   const model = await fromPretrained('mlx-community/parakeet-tdt-0.6b-v3');
+ *   const app = new Hono();
+ *   app.route('/asr', createParakeetRoutes({ model }));
+ *   serve({ fetch: app.fetch, port: 8080 });
+ *
+ * Requires hono as a peer dependency: npm i hono @hono/node-server
+ */
+import { Hono } from 'hono';
+import type { ParakeetModel } from '../model.js';
+export interface ParakeetRouteOptions {
+    model: ParakeetModel;
+    /** Maximum duration of audio accepted per request (default: 300 s). */
+    maxDurationSeconds?: number;
+    /** Close request if no bytes arrive within this window (default: 30 000 ms). */
+    idleTimeoutMs?: number;
+}
+/**
+ * Create a Hono sub-application with the Parakeet ASR routes.
+ * Mount it with `app.route('/asr', createParakeetRoutes({ model }))`.
+ *
+ * Routes:
+ *   POST /transcribe   — stream raw PCM, get AlignedResult JSON
+ */
+export declare function createParakeetRoutes(options: ParakeetRouteOptions): Hono;
+//# sourceMappingURL=server.d.ts.map

package/dist/mlx/server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/mlx/server.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAG5B,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAIjD,MAAM,WAAW,oBAAoB;IACnC,KAAK,EAAE,aAAa,CAAC;IACrB,uEAAuE;IACvE,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,gFAAgF;IAChF,aAAa,CAAC,EAAE,MAAM,CAAC;CACxB;AAED;;;;;;GAMG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,oBAAoB,GAAG,IAAI,CAoHxE"}

package/dist/mlx/server.js

@@ -0,0 +1,115 @@
+/**
+ * parakeet.ts/mlx/server — Hono route factory for the streaming ASR endpoint.
+ *
+ * Usage:
+ *   import { Hono } from 'hono';
+ *   import { serve } from '@hono/node-server';
+ *   import { fromPretrained } from 'parakeet.ts/mlx';
+ *   import { createParakeetRoutes } from 'parakeet.ts/mlx/server';
+ *
+ *   const model = await fromPretrained('mlx-community/parakeet-tdt-0.6b-v3');
+ *   const app = new Hono();
+ *   app.route('/asr', createParakeetRoutes({ model }));
+ *   serve({ fetch: app.fetch, port: 8080 });
+ *
+ * Requires hono as a peer dependency: npm i hono @hono/node-server
+ */
+import { Hono } from 'hono';
+import { consumePcmStream } from '../model.js';
+const REQUIRED_CONTENT_TYPE = 'audio/pcm; rate=16000; channels=1; format=f32le';
+/**
+ * Create a Hono sub-application with the Parakeet ASR routes.
+ * Mount it with `app.route('/asr', createParakeetRoutes({ model }))`.
+ *
+ * Routes:
+ *   POST /transcribe   — stream raw PCM, get AlignedResult JSON
+ */
+export function createParakeetRoutes(options) {
+    const { model, maxDurationSeconds = 300, idleTimeoutMs = 30_000 } = options;
+    const app = new Hono();
+    // Maximum bytes of audio we'll accept: maxDurationSeconds * 16000 Hz * 4 bytes/sample
+    const maxBytes = maxDurationSeconds * 16_000 * 4;
+    app.post('/transcribe', async (c) => {
+        // Validate Content-Type
+        const contentType = c.req.header('content-type') ?? '';
+        if (contentType.trim().toLowerCase() !== REQUIRED_CONTENT_TYPE) {
+            return c.json({ error: 'unsupported_media_type', message: `Content-Type must be "${REQUIRED_CONTENT_TYPE}"` }, 415);
+        }
+        const streamingSession = model.transcribeStream();
+        let totalBytes = 0;
+        let timedOut = false;
+        let tooLarge = false;
+        async function* bodyToPcm() {
+            const body = c.req.raw.body;
+            if (!body)
+                return;
+            const reader = body.getReader();
+            let leftover = new Uint8Array(0);
+            try {
+                for (;;) {
+                    // Idle timeout: race reader.read() against a timeout promise
+                    let timeoutHandle;
+                    const timeoutPromise = new Promise(resolve => {
+                        timeoutHandle = setTimeout(() => resolve(null), idleTimeoutMs);
+                    });
+                    const readPromise = reader.read();
+                    const raceResult = await Promise.race([readPromise, timeoutPromise]);
+                    clearTimeout(timeoutHandle);
+                    if (raceResult === null) {
+                        timedOut = true;
+                        reader.cancel().catch(() => { });
+                        return;
+                    }
+                    const chunk = raceResult;
+                    if (chunk.done)
+                        break;
+                    const incoming = chunk.value;
+                    totalBytes += incoming.byteLength;
+                    if (totalBytes > maxBytes) {
+                        tooLarge = true;
+                        reader.cancel().catch(() => { });
+                        return;
+                    }
+                    // Combine leftover bytes with new chunk
+                    const combined = new Uint8Array(leftover.byteLength + incoming.byteLength);
+                    combined.set(leftover);
+                    combined.set(incoming, leftover.byteLength);
+                    const floatCount = Math.floor(combined.byteLength / 4);
+                    if (floatCount > 0) {
+                        const usedBytes = floatCount * 4;
+                        yield new Float32Array(combined.buffer, combined.byteOffset, floatCount);
+                        leftover = combined.slice(usedBytes);
+                    }
+                    else {
+                        leftover = combined;
+                    }
+                }
+            }
+            finally {
+                reader.releaseLock();
+            }
+        }
+        try {
+            const result = await consumePcmStream(streamingSession, bodyToPcm());
+            if (tooLarge) {
+                return c.json({ error: 'payload_too_large', message: `Audio exceeds maximum of ${maxDurationSeconds} seconds` }, 413);
+            }
+            if (timedOut) {
+                return c.json({ error: 'idle_timeout', message: `No audio received for ${idleTimeoutMs} ms` }, 408);
+            }
+            return c.json(result, 200);
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            // Client disconnect mid-stream: silent drop
+            if (message.includes('aborted') || message.includes('disconnected')) {
+                console.error('[parakeet-mlx] client disconnected mid-stream');
+                return new Response(null, { status: 499 });
+            }
+            console.error('[parakeet-mlx] internal error:', err);
+            return c.json({ error: 'internal_error', message: 'Unexpected error during transcription' }, 500);
+        }
+    });
+    return app;
+}
+//# sourceMappingURL=server.js.map

package/dist/mlx/server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/mlx/server.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAE5B,OAAO,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAG/C,MAAM,qBAAqB,GAAG,iDAAiD,CAAC;AAUhF;;;;;;GAMG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAA6B;IAChE,MAAM,EAAE,KAAK,EAAE,kBAAkB,GAAG,GAAG,EAAE,aAAa,GAAG,MAAM,EAAE,GAAG,OAAO,CAAC;IAE5E,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IAEvB,sFAAsF;IACtF,MAAM,QAAQ,GAAG,kBAAkB,GAAG,MAAM,GAAG,CAAC,CAAC;IAEjD,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,KAAK,EAAE,CAAU,EAAE,EAAE;QAC3C,wBAAwB;QACxB,MAAM,WAAW,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC;QACvD,IAAI,WAAW,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,KAAK,qBAAqB,EAAE,CAAC;YAC/D,OAAO,CAAC,CAAC,IAAI,CACX,EAAE,KAAK,EAAE,wBAAwB,EAAE,OAAO,EAAE,yBAAyB,qBAAqB,GAAG,EAAE,EAC/F,GAAG,CACJ,CAAC;QACJ,CAAC;QAED,MAAM,gBAAgB,GAAG,KAAK,CAAC,gBAAgB,EAAE,CAAC;QAClD,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,IAAI,QAAQ,GAAG,KAAK,CAAC;QAErB,KAAK,SAAS,CAAC,CAAC,SAAS;YACvB,MAAM,IAAI,GAAsC,CAAC,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC;YAC/D,IAAI,CAAC,IAAI;gBAAE,OAAO;YAElB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC;YAChC,IAAI,QAAQ,GAAG,IAAI,UAAU,CAAC,CAAC,CAAC,CAAC;YAEjC,IAAI,CAAC;gBACH,SAAS,CAAC;oBACR,6DAA6D;oBAC7D,IAAI,aAA4C,CAAC;oBACjD,MAAM,cAAc,GAAG,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;wBACjD,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,aAAa,CAAC,CAAC;oBACjE,CAAC,CAAC,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC;oBAClC,MAAM,UAAU,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC,CAAC;oBAErE,YAAY,CAAC,aAAc,CAAC,CAAC;oBAE7B,IAAI,UAAU,KAAK,IAAI,EAAE,CAAC;wBACxB,QAAQ,GAAG,IAAI,CAAC;wBAChB,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;wBAChC,OAAO;oBACT,CAAC;oBAED,MAAM,KAAK,GAAG,UAAmD,CAAC;oBAClE,IAAI,KAAK,CAAC,IAAI;wBAAE,MAAM;oBAEtB,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAM,CAAC;oBAC9B,UAAU,IAAI,QAAQ,CAAC,UAAU,CAAC;oBAElC,IAAI,UAAU,GAAG,QAAQ,EAAE,CAAC;wBAC1B,QAAQ,GAAG,IAAI,CAAC;wBAChB,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;wBAChC,OAAO;oBACT,CAAC;oBAED,wCAAwC;oBACxC,MAAM,QAAQ,GAAG,IAAI,UAAU,CAAC,QAAQ,CAAC,UAAU,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;oBAC3E,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBACvB,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC;oBAE5C,MAAM,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC;oBACvD,IAAI,UAAU,GAAG,CAAC,EAAE,CAAC;wBACnB,MAAM,SAAS,GAAG,UAAU,GAAG,CAAC,CAAC;wBACjC,MAAM,IAAI,YAAY,CAAC,QAAQ,CAAC,MAAM,EAAE,QAAQ,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;wBACzE,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;oBACvC,CAAC;yBAAM,CAAC;wBACN,QAAQ,GAAG,QAAQ,CAAC;oBACtB,CAAC;gBACH,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,MAAM,CAAC,WAAW,EAAE,CAAC;YACvB,CAAC;QACH,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,gBAAgB,EAAE,SAAS,EAAE,CAAC,CAAC;YAErE,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,CAAC,CAAC,IAAI,CACX,EAAE,KAAK,EAAE,mBAAmB,EAAE,OAAO,EAAE,4BAA4B,kBAAkB,UAAU,EAAE,EACjG,GAAG,CACJ,CAAC;YACJ,CAAC;YAED,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO,CAAC,CAAC,IAAI,CACX,EAAE,KAAK,EAAE,cAAc,EAAE,OAAO,EAAE,yBAAyB,aAAa,KAAK,EAAE,EAC/E,GAAG,CACJ,CAAC;YACJ,CAAC;YAED,OAAO,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;QAE7B,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAEjE,4CAA4C;YAC5C,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC,EAAE,CAAC;gBACpE,OAAO,CAAC,KAAK,CAAC,+CAA+C,CAAC,CAAC;gBAC/D,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;YAC7C,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,GAAG,CAAC,CAAC;YACrD,OAAO,CAAC,CAAC,IAAI,CACX,EAAE,KAAK,EAAE,gBAAgB,EAAE,OAAO,EAAE,uCAAuC,EAAE,EAC7E,GAAG,CACJ,CAAC;QACJ,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/mlx/utils.d.ts

@@ -0,0 +1,19 @@
+/**
+ * Low-level model-loading helpers for the MLX backend.
+ *
+ *  - `loadSafetensors` parses a SafeTensors weight file into a WeightMap.
+ *  - `downloadFromHub` is re-exported from `../hub.ts` for backwards
+ *    compatibility; the implementation is backend-agnostic.
+ *
+ * The public loaders (`fromLocal` / `fromPretrained`) live in `./load.ts`, which
+ * builds a backend-agnostic `ParakeetModel` from these pieces.
+ */
+import type { WeightMap } from './nn.js';
+/**
+ * Parse a safetensors file and return a WeightMap.
+ * The safetensors format: 8 bytes (header length LE uint64) + JSON header + data.
+ * Uses fd-based random access to support files larger than the 2 GiB Buffer limit.
+ */
+export declare function loadSafetensors(filePath: string): WeightMap;
+export { downloadFromHub } from '../hub.js';
+//# sourceMappingURL=utils.d.ts.map

package/dist/mlx/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../src/mlx/utils.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAIH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,SAAS,CAAC;AAczC;;;;GAIG;AACH,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,GAAG,SAAS,CAoD3D;AA2CD,OAAO,EAAE,eAAe,EAAE,MAAM,WAAW,CAAC"}