@genai-fi/nanogpt 0.6.2 → 0.7.0

package/dist/{kernel_funcs_utils-CUxJCg0g.js → kernel_funcs_utils-CwRTFqrc.js}

@@ -1,5 +1,6 @@
-import { h as B, g as G, ak as K, _ as W, al as z, am as V, ac as N, an as F, u as S } from "./index-BAzbokzv.js";
-import { u as O, f as Y, h as v } from "./gpgpu_math-CNslybmD.js";
+import { k as B, j as G, ak as K, a1 as W, al as z, am as V, ac as N, an as F, u as S } from "./index-C0dhsYom.js";
+import { u as O, f as Y } from "./gpgpu_math-DJm3ZTAf.js";
+import { f as v } from "./backend_util-DWiwsi2N.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -22,7 +23,7 @@ function H(t, e) {
 function Z(t, e) {
   return e === 1 ? [t] : H(t, e);
 }
-function ie(t, e) {
+function pe(t, e) {
   if (t === 1)
     return "rc";
   let s = "";
@@ -62,11 +63,11 @@ class q {
     `;
   }
 }
-const T = "if (isnan(x)) return x;", M = "return x;", pe = "return abs(x);", j = "return (x >= 0.0) ? x : (exp(x) - 1.0);", J = T + `
+const T = "if (isnan(x)) return x;", j = "return x;", de = "return abs(x);", M = "return (x >= 0.0) ? x : (exp(x) - 1.0);", J = T + `
   return (x < 0.0) ? 0.0 : x;
 `, Q = T + `
   return (x < 0.0) ? 0.0 : min(6.0, x);
-`, de = "return x;", X = "return 1.0 / (1.0 + exp(-1.0 * x));";
+`, he = "return x;", X = "return 1.0 / (1.0 + exp(-1.0 * x));";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -145,7 +146,7 @@ class ne {
  * limitations under the License.
  * =============================================================================
  */
-const he = `
+const fe = `
   if (isnan(a)) return a;
   if (isnan(b)) return b;
 `;
@@ -180,7 +181,7 @@ class b {
  * limitations under the License.
  * =============================================================================
  */
-const fe = `
+const xe = `
   result.r = isNaN.r ? NAN : result.r;
   result.g = isNaN.g ? NAN : result.g;
   result.b = isNaN.b ? NAN : result.b;
@@ -268,7 +269,7 @@ function P(t) {
   const { inputs: e, backend: s } = t, { x: r } = e;
   return s.incRef(r.dataId), { dataId: r.dataId, shape: r.shape, dtype: r.dtype };
 }
-const xe = {
+const ge = {
   kernelName: K,
   backendName: "webgl",
   kernelFunc: P
@@ -293,7 +294,7 @@ function L(t) {
   const { inputs: e, backend: s } = t, { real: r, imag: u } = e, n = s.makeTensorInfo(r.shape, "complex64"), o = s.texData.get(n.dataId), i = P({ inputs: { x: r }, backend: s }), a = P({ inputs: { x: u }, backend: s });
   return o.complexTensorInfos = { real: i, imag: a }, n;
 }
-const ge = {
+const me = {
   kernelName: W,
   backendName: "webgl",
   kernelFunc: L
@@ -322,7 +323,7 @@ function oe(t) {
   const { inputs: e, backend: s, attrs: r } = t, { x: u } = e, { alpha: n } = r, o = s.makeTensorInfo([], "float32", V(n, "float32")), i = N().getBool("WEBGL_PACK_BINARY_OPERATIONS") ? new E(k, u.shape, o.shape) : new b(w, u.shape, o.shape), a = s.runWebGLProgram(i, [u, o], "float32");
   return s.disposeIntermediateTensorInfo(o), a;
 }
-const me = {
+const be = {
   kernelName: z,
   backendName: "webgl",
   kernelFunc: oe
@@ -351,7 +352,7 @@ function ue(t) {
   const { inputs: e, backend: s } = t, { x: r, alpha: u } = e, n = N().getBool("WEBGL_PACK_BINARY_OPERATIONS") ? new E(U, r.shape, u.shape) : new b(R, r.shape, u.shape);
   return s.runWebGLProgram(n, [r, u], "float32");
 }
-const be = {
+const Ne = {
   kernelName: F,
   backendName: "webgl",
   kernelFunc: ue
@@ -372,8 +373,8 @@ const be = {
  * limitations under the License.
  * =============================================================================
  */
-const Ne = "if (isnan(x)) return x;";
-function Oe({ opSnippet: t, packedOpSnippet: e, cpuKernelImpl: s, dtype: r }) {
+const Oe = "if (isnan(x)) return x;";
+function ye({ opSnippet: t, packedOpSnippet: e, cpuKernelImpl: s, dtype: r }) {
   return ({ inputs: u, backend: n }) => {
     const { x: o } = u, i = n, a = r || o.dtype;
     if (i.shouldExecuteOnCPU([o]) && s != null) {
@@ -385,7 +386,7 @@ function Oe({ opSnippet: t, packedOpSnippet: e, cpuKernelImpl: s, dtype: r }) {
     return c ? l = new ne(o.shape, e) : l = new q(o.shape, t), i.runWebGLProgram(l, [o], a);
   };
 }
-function ye({ opSnippet: t, packedOpSnippet: e, checkOutOfBounds: s = !1, supportsComplex: r = !1, cpuKernelImpl: u, dtype: n }) {
+function Ie({ opSnippet: t, packedOpSnippet: e, checkOutOfBounds: s = !1, supportsComplex: r = !1, cpuKernelImpl: u, dtype: n }) {
   return ({ inputs: o, backend: i }) => {
     const { a, b: c } = o, l = i;
     if (r && a.dtype === "complex64") {
@@ -422,13 +423,13 @@ function ye({ opSnippet: t, packedOpSnippet: e, checkOutOfBounds: s = !1, suppor
     return y ? I = new E(e, a.shape, c.shape, s) : I = new b(t, a.shape, c.shape), l.runWebGLProgram(I, [a, c], d);
   };
 }
-function Ie(t, e = !1) {
+function Ae(t, e = !1) {
   if (t === "linear")
-    return e ? ee : M;
+    return e ? ee : j;
   if (t === "relu")
     return e ? se : J;
   if (t === "elu")
-    return e ? te : j;
+    return e ? te : M;
   if (t === "relu6")
     return e ? ae : Q;
   if (t === "prelu")
@@ -440,26 +441,26 @@ function Ie(t, e = !1) {
   throw new Error(`Activation ${t} has not been implemented for the WebGL backend.`);
 }
 export {
-  pe as A,
+  de as A,
   E as B,
   T as C,
   ne as U,
   Z as a,
-  ye as b,
-  ie as c,
-  de as d,
+  Ie as b,
+  pe as c,
+  he as d,
   q as e,
   L as f,
   H as g,
   b as h,
   P as i,
-  he as j,
-  fe as k,
-  Ne as l,
-  Ie as m,
-  ge as n,
-  xe as o,
-  me as p,
-  be as q,
-  Oe as u
+  fe as j,
+  xe as k,
+  Oe as l,
+  Ae as m,
+  me as n,
+  ge as o,
+  be as p,
+  Ne as q,
+  ye as u
 };

package/dist/layers/BaseLayer.js

@@ -1,5 +1,5 @@
-import { T as g, l as p, e as o, n as v } from "../index-BAzbokzv.js";
-import { v as _ } from "../variable-wSS22xj5.js";
+import { T as g, q as p, e as o, w as v } from "../index-C0dhsYom.js";
+import { v as _ } from "../variable-DXEUOwew.js";
 class M {
   parent;
   config;

package/dist/layers/CausalSelfAttention.js

@@ -3,13 +3,13 @@ import O from "./BaseLayer.js";
 import { qkv as P } from "../ops/qkv.js";
 import { rope as v } from "../ops/rope.js";
 import { appendCache as V } from "../ops/appendCache.js";
-import { k as c, t as C } from "../index-BAzbokzv.js";
+import { o as c, t as C } from "../index-C0dhsYom.js";
 import { fusedSoftmax as T } from "../ops/fusedSoftmax.js";
-import { d as y } from "../random_width-DWzaOgrn.js";
-import { v as b } from "../variable-wSS22xj5.js";
-import { r as k, d as L } from "../dropout-Bciw46HT.js";
-import { r as N } from "../reshape-krWGKraP.js";
-import { m as R } from "../mat_mul-7121rsJk.js";
+import { d as y } from "../random_width-DkYP8W8N.js";
+import { v as b } from "../variable-DXEUOwew.js";
+import { r as k, d as L } from "../dropout-hpDwECTe.js";
+import { r as N } from "../reshape-BMUzc1UY.js";
+import { m as R } from "../mat_mul-1nwdPkQ_.js";
 class $ extends O {
   divisor;
   index;

package/dist/layers/MLP.js

@@ -1,10 +1,10 @@
-import { t as l } from "../index-BAzbokzv.js";
+import { t as l } from "../index-C0dhsYom.js";
 import u from "./BaseLayer.js";
 import { matMulGelu as M } from "../ops/matMulGelu.js";
-import { v as o } from "../variable-wSS22xj5.js";
-import { r as h, d as f } from "../dropout-Bciw46HT.js";
-import { r as d } from "../reshape-krWGKraP.js";
-import { m as c } from "../mat_mul-7121rsJk.js";
+import { v as o } from "../variable-DXEUOwew.js";
+import { r as h, d as f } from "../dropout-hpDwECTe.js";
+import { r as d } from "../reshape-BMUzc1UY.js";
+import { m as c } from "../mat_mul-1nwdPkQ_.js";
 class V extends u {
   index;
   hiddenUnits;

package/dist/layers/RMSNorm.js

@@ -1,8 +1,8 @@
-import { t as s } from "../index-BAzbokzv.js";
+import { t as s } from "../index-C0dhsYom.js";
 import e from "./BaseLayer.js";
 import { normRMS as a } from "../ops/normRMS.js";
-import { v as i } from "../variable-wSS22xj5.js";
-import { o as m } from "../ones-BBlSRqn1.js";
+import { v as i } from "../variable-DXEUOwew.js";
+import { o as m } from "../ones-D63HpSF_.js";
 class f extends e {
   GAMMA;
   constructor(r, t = "", o) {

package/dist/layers/RoPECache.js

@@ -1,7 +1,7 @@
-import { s as t, j as h, t as n, k as p } from "../index-BAzbokzv.js";
-import { r as c } from "../reciprocal-CJQeasVa.js";
-import { c as f, s as m } from "../sin-D_h-qCSx.js";
-import { r as a } from "../range-DYsrnfiy.js";
+import { b as t, n as h, t as n, o as p } from "../index-C0dhsYom.js";
+import { r as c } from "../reciprocal-_A9yv27J.js";
+import { c as f, s as m } from "../sin-Cocju-BY.js";
+import { r as a } from "../range-CYzpQY53.js";
 class D {
   constructor(o) {
     this.config = o;

package/dist/layers/TiedEmbedding.js

@@ -1,9 +1,9 @@
-import "../random_width-DWzaOgrn.js";
-import "../index-BAzbokzv.js";
-import { T as e } from "../TiedEmbedding-9WeDwvjO.js";
+import "../random_width-DkYP8W8N.js";
+import "../index-C0dhsYom.js";
+import { T as e } from "../TiedEmbedding-BLltddza.js";
 import "./BaseLayer.js";
-import "../variable-wSS22xj5.js";
-import "../gather-DjyCjmOD.js";
+import "../variable-DXEUOwew.js";
+import "../gather-D0_gPiBz.js";
 export {
   e as default
 };

package/dist/layers/TransformerBlock.js

@@ -2,7 +2,7 @@ import l from "./CausalSelfAttention.js";
 import r from "./MLP.js";
 import o from "./RMSNorm.js";
 import d from "./BaseLayer.js";
-import { t as p } from "../index-BAzbokzv.js";
+import { t as p } from "../index-C0dhsYom.js";
 class k extends d {
   ln1;
   attn;

package/dist/loader/loadTransformers.js

@@ -2,7 +2,7 @@ import b from "../NanoGPTModel.js";
 import c from "../tokeniser/CharTokeniser.js";
 import l from "../tokeniser/bpe.js";
 import { load_safetensors as u } from "../utilities/safetensors.js";
-import { U as y } from "../index-BAzbokzv.js";
+import { Y as y } from "../index-C0dhsYom.js";
 import { dummyPassAsync as h } from "../utilities/dummy.js";
 async function L(e, a, r, t) {
   const n = {

package/dist/loader/oldZipLoad.js

@@ -1,12 +1,12 @@
 import d from "../NanoGPTModel.js";
 import "../jszip.min-CjP2V1VV.js";
 import h from "../tokeniser/CharTokeniser.js";
-import { U as k } from "../index-BAzbokzv.js";
+import { Y as k } from "../index-C0dhsYom.js";
 import b from "../tokeniser/bpe.js";
 import { dummyPassAsync as u } from "../utilities/dummy.js";
 import "../Generator.js";
 import "../index-Dwqa6Zy2.js";
-import "../dataset-pgqp-YfL.js";
+import "../dataset-JDyjG3QR.js";
 import "../index-Tf7vU29b.js";
 import "../papaparse.min-C8l2Kvo1.js";
 import "../ops/cpu/scatterSub.js";
@@ -19,8 +19,8 @@ import "../ops/grads/attentionMask.js";
 import "../ops/cpu/qkv.js";
 import "../ops/webgl/qkv.js";
 import "../ops/grads/qkv.js";
-import "../random_width-DWzaOgrn.js";
-import "../register_all_kernels-BfFCQAqs.js";
+import "../random_width-DkYP8W8N.js";
+import "../register_all_kernels-guvSxp7M.js";
 import "../ops/cpu/rope.js";
 import "../ops/webgl/rope.js";
 import "../ops/grads/rope.js";
@@ -34,7 +34,7 @@ import "../ops/webgl/matMulGelu.js";
 import "../ops/grads/matMulGelu.js";
 import "../ops/cpu/gelu.js";
 import "../ops/webgl/gelu.js";
-import "../ops/grads/gelu.js";
+import "../gelu-uyHP1x1f.js";
 import "../ops/cpu/normRMS.js";
 import "../ops/webgl/normRMS.js";
 import "../ops/grads/normRMS.js";

package/dist/{log_sum_exp-YEo2h3gb.js → log_sum_exp-D086OgZJ.js}

@@ -1,8 +1,8 @@
-import { o as r, q as p, E as u, a8 as E, a9 as h, p as S, b as $, a7 as d } from "./index-BAzbokzv.js";
-import { e as b } from "./axis_util-Bu4h7XWV.js";
-import { m as K } from "./max-DtlIuVeW.js";
-import { r as m } from "./reshape-krWGKraP.js";
-import { s as T } from "./sum-BE_Irnim.js";
+import { x as r, y as p, E as u, a7 as E, a8 as h, p as S, c as $, a6 as d } from "./index-C0dhsYom.js";
+import { e as K } from "./axis_util-DaAl5MER.js";
+import { m as T } from "./max-BQc2Aj-I.js";
+import { r as m } from "./reshape-BMUzc1UY.js";
+import { s as _ } from "./sum-B8wEpKsg.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -19,11 +19,11 @@ import { s as T } from "./sum-BE_Irnim.js";
  * limitations under the License.
  * =============================================================================
  */
-function _(s) {
+function b(s) {
   const o = { x: p(s, "x", "exp") };
   return u.runKernel(E, o);
 }
-const N = /* @__PURE__ */ r({ exp_: _ });
+const N = /* @__PURE__ */ r({ exp_: b });
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -40,11 +40,11 @@ const N = /* @__PURE__ */ r({ exp_: _ });
  * limitations under the License.
  * =============================================================================
  */
-function q(s) {
+function v(s) {
   const o = { x: p(s, "x", "log", "float32") };
   return u.runKernel(h, o);
 }
-const v = /* @__PURE__ */ r({ log_: q });
+const w = /* @__PURE__ */ r({ log_: v });
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -61,22 +61,22 @@ const v = /* @__PURE__ */ r({ log_: q });
  * limitations under the License.
  * =============================================================================
  */
-function w(s, n = null, o = !1) {
-  const a = p(s, "x", "logSumExp"), t = S(n, a.shape), x = K(
+function y(s, n = null, o = !1) {
+  const a = p(s, "x", "logSumExp"), t = S(n, a.shape), x = T(
     a,
     t,
     !0
     /* keepDims */
-  ), i = $(a, x), l = N(i), f = T(l, t), c = v(f), e = d(m(x, c.shape), c);
+  ), i = $(a, x), l = N(i), f = _(l, t), c = w(f), e = d(m(x, c.shape), c);
   if (o) {
-    const g = b(e.shape, t);
+    const g = K(e.shape, t);
     return m(e, g);
   }
   return e;
 }
-const M = /* @__PURE__ */ r({ logSumExp_: w });
+const M = /* @__PURE__ */ r({ logSumExp_: y });
 export {
-  v as a,
+  w as a,
   N as e,
   M as l
 };

package/dist/main.d.ts

@@ -11,3 +11,5 @@ export type { ITokeniser } from './tokeniser/type';
 export type { TrainingProgress } from './training/Trainer';
 export type { GPTConfig } from './config';
 export { estimateParameterCount, estimateMemoryUsage, estimateTrainingMemoryUsage, estimateResources, validateConfig, } from './utilities/parameters';
+export { selectBackend } from './backend';
+export { default as performanceTest } from './utilities/performance';

package/dist/main.js

@@ -5,7 +5,7 @@ import { default as I } from "./tokeniser/bpe.js";
 import { default as K } from "./utilities/waitForModel.js";
 import { default as Q } from "./data/textLoader.js";
 import { estimateMemoryUsage as V, estimateParameterCount as W, estimateResources as X, estimateTrainingMemoryUsage as Y, validateConfig as Z } from "./utilities/parameters.js";
-import "./index-BAzbokzv.js";
+import "./index-C0dhsYom.js";
 import "./ops/cpu/scatterSub.js";
 import "./ops/webgl/scatterSub.js";
 import "./ops/cpu/gatherSub.js";
@@ -16,10 +16,10 @@ import "./ops/grads/attentionMask.js";
 import "./ops/cpu/qkv.js";
 import "./ops/webgl/qkv.js";
 import "./ops/grads/qkv.js";
-import "./random_width-DWzaOgrn.js";
-import "./register_all_kernels-BfFCQAqs.js";
+import "./random_width-DkYP8W8N.js";
+import "./register_all_kernels-guvSxp7M.js";
 import "./index-Tf7vU29b.js";
-import "./dataset-pgqp-YfL.js";
+import "./dataset-JDyjG3QR.js";
 import "./ops/cpu/rope.js";
 import "./ops/webgl/rope.js";
 import "./ops/grads/rope.js";
@@ -33,11 +33,13 @@ import "./ops/webgl/matMulGelu.js";
 import "./ops/grads/matMulGelu.js";
 import "./ops/cpu/gelu.js";
 import "./ops/webgl/gelu.js";
-import "./ops/grads/gelu.js";
+import "./gelu-uyHP1x1f.js";
 import "./ops/cpu/normRMS.js";
 import "./ops/webgl/normRMS.js";
 import "./ops/grads/normRMS.js";
 import "./ops/webgl/log.js";
+import { selectBackend as $ } from "./backend.js";
+import { default as ot } from "./utilities/performance.js";
 export {
   I as BPETokeniser,
   A as CharTokeniser,
@@ -48,6 +50,8 @@ export {
   X as estimateResources,
   Y as estimateTrainingMemoryUsage,
   Q as loadTextData,
+  ot as performanceTest,
+  $ as selectBackend,
   Z as validateConfig,
   K as waitForModel
 };

package/dist/{mat_mul-7121rsJk.js → mat_mul-1nwdPkQ_.js}

@@ -1,4 +1,4 @@
-import { o as m, q as s, B as c, E as M, C as p } from "./index-BAzbokzv.js";
+import { x as m, y as s, F as c, E as M, H as p } from "./index-C0dhsYom.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -15,10 +15,10 @@ import { o as m, q as s, B as c, E as M, C as p } from "./index-BAzbokzv.js";
  * limitations under the License.
  * =============================================================================
  */
-function f(e, o, n = !1, l = !1) {
-  let a = s(e, "a", "matMul"), t = s(o, "b", "matMul");
+function f(e, n, o = !1, l = !1) {
+  let a = s(e, "a", "matMul"), t = s(n, "b", "matMul");
   [a, t] = c(a, t);
-  const r = { a, b: t }, u = { transposeA: n, transposeB: l };
+  const r = { a, b: t }, u = { transposeA: o, transposeB: l };
   return M.runKernel(p, r, u);
 }
 const i = /* @__PURE__ */ m({ matMul_: f });

package/dist/{max-DtlIuVeW.js → max-BQc2Aj-I.js}

@@ -1,4 +1,4 @@
-import { o as r, q as e, E as x, M as c } from "./index-BAzbokzv.js";
+import { x as r, y as x, E as e, N as c } from "./index-C0dhsYom.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -15,9 +15,9 @@ import { o as r, q as e, E as x, M as c } from "./index-BAzbokzv.js";
  * limitations under the License.
  * =============================================================================
  */
-function m(n, o = null, s = !1) {
-  const t = { x: e(n, "x", "max") }, a = { reductionIndices: o, keepDims: s };
-  return x.runKernel(c, t, a);
+function m(n, s = null, o = !1) {
+  const t = { x: x(n, "x", "max") }, a = { reductionIndices: s, keepDims: o };
+  return e.runKernel(c, t, a);
 }
 const l = /* @__PURE__ */ r({ max_: m });
 export {

package/dist/{mulmat_packed_gpu-D4nKF7Je.js → mulmat_packed_gpu-Gzf3I9UV.js}

@@ -1,4 +1,4 @@
-import { u as z } from "./gpgpu_math-CNslybmD.js";
+import { u as z } from "./gpgpu_math-DJm3ZTAf.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/non_max_suppression_impl-CsEgBuMA.js

@@ -0,0 +1,134 @@
+/**
+ * @license
+ * Copyright 2019 Google LLC. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * =============================================================================
+ */
+function J(e, t, o) {
+  const n = k(e, t, o), s = n < 0 ? -(n + 1) : n;
+  e.splice(s, 0, t);
+}
+function k(e, t, o) {
+  return A(e, t, o || v);
+}
+function v(e, t) {
+  return e > t ? 1 : e < t ? -1 : 0;
+}
+function A(e, t, o) {
+  let n = 0, s = e.length, r = 0, u = !1;
+  for (; n < s; ) {
+    r = n + (s - n >>> 1);
+    const l = o(t, e[r]);
+    l > 0 ? n = r + 1 : (s = r, u = !l);
+  }
+  return u ? n : -n - 1;
+}
+/**
+ * @license
+ * Copyright 2020 Google LLC. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * =============================================================================
+ */
+function O(e, t, o, n, s) {
+  return y(
+    e,
+    t,
+    o,
+    n,
+    s,
+    0
+    /* softNmsSigma */
+  );
+}
+function P(e, t, o, n, s, r) {
+  return y(
+    e,
+    t,
+    o,
+    n,
+    s,
+    0,
+    !1,
+    r,
+    !0
+    /* returnValidOutputs */
+  );
+}
+function X(e, t, o, n, s, r) {
+  return y(
+    e,
+    t,
+    o,
+    n,
+    s,
+    r,
+    !0
+    /* returnScoresTensor */
+  );
+}
+function y(e, t, o, n, s, r, u = !1, l = !1, h = !1) {
+  const c = [];
+  for (let i = 0; i < t.length; i++)
+    t[i] > s && c.push({ score: t[i], boxIndex: i, suppressBeginIndex: 0 });
+  c.sort(C);
+  const d = r > 0 ? -0.5 / r : 0, a = [], f = [];
+  for (; a.length < o && c.length > 0; ) {
+    const i = c.pop(), { score: M, boxIndex: I, suppressBeginIndex: g } = i;
+    if (M < s)
+      break;
+    let S = !1;
+    for (let b = a.length - 1; b >= g; --b) {
+      const w = B(e, I, a[b]);
+      if (w >= n) {
+        S = !0;
+        break;
+      }
+      if (i.score = i.score * V(n, d, w), i.score <= s)
+        break;
+    }
+    i.suppressBeginIndex = a.length, S || (i.score === M ? (a.push(I), f.push(i.score)) : i.score > s && J(c, i, C));
+  }
+  const x = a.length, p = o - x;
+  l && p > 0 && (a.push(...new Array(p).fill(0)), f.push(...new Array(p).fill(0)));
+  const m = { selectedIndices: a };
+  return u && (m.selectedScores = f), h && (m.validOutputs = x), m;
+}
+function B(e, t, o) {
+  const n = e.subarray(t * 4, t * 4 + 4), s = e.subarray(o * 4, o * 4 + 4), r = Math.min(n[0], n[2]), u = Math.min(n[1], n[3]), l = Math.max(n[0], n[2]), h = Math.max(n[1], n[3]), c = Math.min(s[0], s[2]), d = Math.min(s[1], s[3]), a = Math.max(s[0], s[2]), f = Math.max(s[1], s[3]), x = (l - r) * (h - u), p = (a - c) * (f - d);
+  if (x <= 0 || p <= 0)
+    return 0;
+  const m = Math.max(r, c), i = Math.max(u, d), M = Math.min(l, a), I = Math.min(h, f), g = Math.max(M - m, 0) * Math.max(I - i, 0);
+  return g / (x + p - g);
+}
+function V(e, t, o) {
+  const n = Math.exp(t * o * o);
+  return o <= e ? n : 0;
+}
+function C(e, t) {
+  return e.score - t.score || e.score === t.score && t.boxIndex - e.boxIndex;
+}
+export {
+  X as a,
+  P as b,
+  O as n
+};

package/dist/{ones-BBlSRqn1.js → ones-D63HpSF_.js}

@@ -1,5 +1,5 @@
-import { x as n, y as t, g as m, E as i } from "./index-BAzbokzv.js";
-import { z as c, c as f } from "./zeros-YJDE7oRb.js";
+import { C as n, D as t, j as m, E as i } from "./index-C0dhsYom.js";
+import { z as c, c as f } from "./zeros-DCPCdFGq.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/ops/appendCache.js

@@ -1,8 +1,8 @@
-import { e as a } from "../index-BAzbokzv.js";
+import { e as a } from "../index-C0dhsYom.js";
 import "./cpu/appendCache.js";
 import "./webgl/appendCache.js";
-import { c as s } from "../concat-5aPGqw3Z.js";
-import { z as c } from "../zeros-YJDE7oRb.js";
+import { c as s } from "../concat-CsHeR4zV.js";
+import { z as c } from "../zeros-DCPCdFGq.js";
 function i(r, p, n, o) {
   if (!o) {
     const e = r.shape[2];

package/dist/ops/attentionMask.js

@@ -1,4 +1,4 @@
-import { e as o } from "../index-BAzbokzv.js";
+import { e as o } from "../index-C0dhsYom.js";
 import "./cpu/attentionMask.js";
 import "./webgl/attentionMask.js";
 import "./grads/attentionMask.js";