@genai-fi/nanogpt 0.7.0 → 0.7.1

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

@@ -1,5 +1,5 @@
-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";
+import { F as n, H as t, j as m, E as i } from "./index-UdZhlibC.js";
+import { z as c, c as f } from "./zeros-DnPT2nD4.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/ops/adamAdjust.d.ts

@@ -0,0 +1,2 @@
+import { Tensor } from '@tensorflow/tfjs-core';
+export declare function adamAdjust(moments: Tensor, value: Tensor, beta1: number, beta2: number, epsilon: number, learningRate: number): Tensor;

package/dist/ops/adamAdjust.js

@@ -0,0 +1,9 @@
+import { e as i } from "../index-UdZhlibC.js";
+import "./cpu/adamAdjust.js";
+import "./webgl/adamAdjust.js";
+function p(r, t, e, n, m, o) {
+  return i().runKernel("AdamAdjust", { moments: r, value: t }, { beta1: e, beta2: n, epsilon: m, learningRate: o });
+}
+export {
+  p as adamAdjust
+};

package/dist/ops/adamMoments.d.ts

@@ -0,0 +1,2 @@
+import { Tensor } from '@tensorflow/tfjs-core';
+export declare function adamMoments(moments: Tensor, gradient: Tensor, beta1: number, beta2: number): Tensor;

package/dist/ops/adamMoments.js

@@ -0,0 +1,9 @@
+import { e as o } from "../index-UdZhlibC.js";
+import "./cpu/adamMoments.js";
+import "./webgl/adamMoments.js";
+function p(e, n, r, m) {
+  return o().runKernel("AdamMoments", { moments: e, gradient: n }, { beta1: r, beta2: m });
+}
+export {
+  p as adamMoments
+};

package/dist/ops/appendCache.js

@@ -1,8 +1,8 @@
-import { e as a } from "../index-C0dhsYom.js";
+import { e as a } from "../index-UdZhlibC.js";
 import "./cpu/appendCache.js";
 import "./webgl/appendCache.js";
-import { c as s } from "../concat-CsHeR4zV.js";
-import { z as c } from "../zeros-DCPCdFGq.js";
+import { c as s } from "../concat-CbXTetof.js";
+import { z as c } from "../zeros-DnPT2nD4.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-C0dhsYom.js";
+import { e as o } from "../index-UdZhlibC.js";
 import "./cpu/attentionMask.js";
 import "./webgl/attentionMask.js";
 import "./grads/attentionMask.js";

package/dist/ops/cpu/adamAdjust.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/ops/cpu/adamAdjust.js

@@ -0,0 +1,18 @@
+import { f as k, n as t, o as i, a as q, q as z } from "../../index-UdZhlibC.js";
+function A(a) {
+  const { moments: n, value: r } = a.inputs, { beta1: l, beta2: u, epsilon: m, learningRate: d } = a.attrs, e = n.shape.length, c = new Array(e).fill(0), s = n.shape.slice();
+  s[e - 1] = 1;
+  const o = c.slice();
+  o[e - 1] = 1;
+  const b = s.slice(), p = n.slice(c, s).squeeze([e - 1]), M = n.slice(o, b).squeeze([e - 1]), f = t(p, l), g = t(M, u);
+  return i(
+    q(t(f, i(z(g), m ?? 1e-8)), -d),
+    r
+  );
+}
+const C = {
+  kernelName: "AdamAdjust",
+  backendName: "cpu",
+  kernelFunc: A
+};
+k(C);

package/dist/ops/cpu/adamMoments.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/ops/cpu/adamMoments.js

@@ -0,0 +1,16 @@
+import { f as p } from "../../index-UdZhlibC.js";
+import { s as f } from "../../stack-ikk2Y8_P.js";
+function b(t) {
+  const { moments: n, gradient: c } = t.inputs, { beta1: o, beta2: m } = t.attrs, e = n.shape.length, a = new Array(e).fill(0), s = n.shape.slice();
+  s[e - 1] = 1;
+  const i = a.slice();
+  i[e - 1] = 1;
+  const r = s.slice(), l = n.slice(a, s).squeeze([e - 1]), u = n.slice(i, r).squeeze([e - 1]), M = l.mul(o).add(c.mul(1 - o)), d = u.mul(m).add(c.square().mul(1 - m));
+  return f([M, d], -1);
+}
+const g = {
+  kernelName: "AdamMoments",
+  backendName: "cpu",
+  kernelFunc: b
+};
+p(g);

package/dist/ops/cpu/appendCache.js

@@ -1,5 +1,5 @@
-import { f as d } from "../../index-C0dhsYom.js";
-import { c as h } from "../../concat-CsHeR4zV.js";
+import { f as d } from "../../index-UdZhlibC.js";
+import { c as h } from "../../concat-CbXTetof.js";
 function u(p) {
   const { cache: n, item: s } = p.inputs, { maxSize: i, pastLen: c } = p.attrs, t = n.shape[0], o = n.shape[1], a = n.shape[3], e = s.shape[2];
   if (c + e <= i) {

package/dist/ops/cpu/attentionMask.js

@@ -1,8 +1,8 @@
-import { f as a, h as p, b as u } from "../../index-C0dhsYom.js";
-import { l as N, w as b } from "../../ops-CIQLNshk.js";
-import { o as A } from "../../ones-D63HpSF_.js";
-import { z as I } from "../../zeros-DCPCdFGq.js";
-import { m as g } from "../../mat_mul-1nwdPkQ_.js";
+import { f as a, h as p, b as u } from "../../index-UdZhlibC.js";
+import { l as N, w as b } from "../../ops-aRTXR2Sr.js";
+import { o as A } from "../../ones-C8Mfln6-.js";
+import { z as I } from "../../zeros-DnPT2nD4.js";
+import { m as g } from "../../mat_mul-dwmZz69e.js";
 function o(n) {
   const { q: s, k: e } = n.inputs, { divisor: r } = n.attrs, c = s.shape[2], t = e.shape[2], m = N.bandPart(A([t, t]), -1, 0).cast("bool"), l = I([t, t]), i = p([t, t], Number.NEGATIVE_INFINITY), f = b(m, l, i), k = g(s, e, !1, !0).mul(u(r)), d = f.slice([0, 0], [c, t]).expandDims(0).expandDims(0);
   return k.add(d);

package/dist/ops/cpu/fusedSoftmax.js

@@ -1,5 +1,5 @@
-import { f as e } from "../../index-C0dhsYom.js";
-import { s as f } from "../../softmax-GPNK3o-U.js";
+import { f as e } from "../../index-UdZhlibC.js";
+import { s as f } from "../../softmax-JMEIUo2J.js";
 function n(t) {
   const { inputs: s, attrs: a } = t, { logits: o } = s, { dim: i, dropoutRate: r } = a;
   if (!o)

package/dist/ops/cpu/gatherSub.js

@@ -1,6 +1,6 @@
-import { x as u, y as c, E as g, $ as h, f as m, c as p } from "../../index-C0dhsYom.js";
-import { r as f } from "../../range-CYzpQY53.js";
-import { s as l } from "../../stack-Dpgg_1W1.js";
+import { B as u, C as c, E as g, a4 as h, f as m, c as p } from "../../index-UdZhlibC.js";
+import { r as f } from "../../range-D9CZhVlR.js";
+import { s as l } from "../../stack-ikk2Y8_P.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/ops/cpu/gelu.js

@@ -1,4 +1,4 @@
-import { f as t, t as d } from "../../index-C0dhsYom.js";
+import { f as t, t as d } from "../../index-UdZhlibC.js";
 const o = 0.7978845608028654, c = 0.044715;
 function m(r) {
   const { inputs: u } = r, { x: n } = u, e = n;

package/dist/ops/cpu/matMulGelu.js

@@ -1,5 +1,5 @@
-import { f as e, t as m } from "../../index-C0dhsYom.js";
-import { g as d, d as M } from "../../gelu-uyHP1x1f.js";
+import { f as e, t as m } from "../../index-UdZhlibC.js";
+import { g as d, d as M } from "../../gelu-BFwVnd1r.js";
 function c(t) {
   const { inputs: u } = t, { x: n, kernel: l } = u, a = n, r = l;
   return m(() => {

package/dist/ops/cpu/matMulMul.js

@@ -1,4 +1,4 @@
-import { f as e, t as i } from "../../index-C0dhsYom.js";
+import { f as e, t as i } from "../../index-UdZhlibC.js";
 function n(t) {
   const { inputs: r, attrs: l } = t, { transposeA: o, transposeB: s } = l, { x: u, kernel: a, y: c } = r, m = u, k = a, M = c;
   return i(() => m.matMul(k, o, s).mul(M));

package/dist/ops/cpu/mulDropout.js

@@ -1,4 +1,4 @@
-import { f as e, a as t } from "../../index-C0dhsYom.js";
+import { f as e, a as t } from "../../index-UdZhlibC.js";
 function n(o) {
   const { inputs: r } = o, { a: l, b: u } = r;
   return console.warn("Using fallback mulDrop implementation without dropout."), t(l, u);

package/dist/ops/cpu/normRMS.js

@@ -1,4 +1,4 @@
-import { f as o, t as d } from "../../index-C0dhsYom.js";
+import { f as o, t as d } from "../../index-UdZhlibC.js";
 function i(t) {
   const { inputs: e } = t, { x: n, gamma: s } = e, r = n, a = s;
   return d(() => {

package/dist/ops/cpu/qkv.js

@@ -1,6 +1,6 @@
-import { f as q } from "../../index-C0dhsYom.js";
-import { r as o } from "../../reshape-BMUzc1UY.js";
-import { s as x } from "../../split-CHzJjxDv.js";
+import { f as q } from "../../index-UdZhlibC.js";
+import { r as o } from "../../reshape-BR0eoLYN.js";
+import { s as x } from "../../split-CRU0PjVV.js";
 function v(p) {
   const { x: c, kernel: K } = p.inputs, { heads: n } = p.attrs, [s, e, t] = c.shape, a = o(c, [s * e, t]), i = a.dot(K);
   a.dispose();

package/dist/ops/cpu/rope.js

@@ -1,8 +1,8 @@
-import { f as S } from "../../index-C0dhsYom.js";
-import { r as F } from "../../range-CYzpQY53.js";
-import { g as I } from "../../gather-D0_gPiBz.js";
-import { s as E } from "../../stack-Dpgg_1W1.js";
-import { c as T } from "../../concat-CsHeR4zV.js";
+import { f as S } from "../../index-UdZhlibC.js";
+import { r as F } from "../../range-D9CZhVlR.js";
+import { g as I } from "../../gather-Bbh8DHhM.js";
+import { s as E } from "../../stack-ikk2Y8_P.js";
+import { c as T } from "../../concat-CbXTetof.js";
 function U(t, c, p, o, r) {
   const n = o.shape[3], s = p;
   if (s > n) return o;

package/dist/ops/cpu/scatterSub.js

@@ -1,8 +1,8 @@
-import { x as f, C as g, y as r, E as l, _ as N, f as b, c as S, a as h } from "../../index-C0dhsYom.js";
-import { v as D } from "../../scatter_nd_util-IRBqKz_b.js";
-import { r as k } from "../../range-CYzpQY53.js";
-import { s as v } from "../../stack-Dpgg_1W1.js";
-import { o as E } from "../../ones-D63HpSF_.js";
+import { B as f, F as g, C as r, E as l, a3 as N, f as b, c as S, a as h } from "../../index-UdZhlibC.js";
+import { v as D } from "../../scatter_nd_util-OjyAxku2.js";
+import { r as k } from "../../range-D9CZhVlR.js";
+import { s as v } from "../../stack-ikk2Y8_P.js";
+import { o as E } from "../../ones-C8Mfln6-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -26,14 +26,14 @@ function I(a, e, s) {
   const c = { indices: n, updates: t }, o = { shape: s };
   return l.runKernel(N, c, o);
 }
-const y = /* @__PURE__ */ f({ scatterND_: I });
-function C(a) {
-  const { logits: e, labels: s, dy: n } = a.inputs, t = s.shape[0], c = e.shape[1], o = k(0, t, 1, "int32"), i = v([o, s], 1), d = E([t]), u = y(i, d, [t, c]), p = S(e, u), m = n.reshape([t, 1]);
+const C = /* @__PURE__ */ f({ scatterND_: I });
+function K(a) {
+  const { logits: e, labels: s, dy: n } = a.inputs, t = s.shape[0], c = e.shape[1], o = k(0, t, 1, "int32"), i = v([o, s], 1), d = E([t]), u = C(i, d, [t, c]), p = S(e, u), m = n.reshape([t, 1]);
   return h(p, m);
 }
-const K = {
+const L = {
   kernelName: "EfficientScatterSub",
   backendName: "cpu",
-  kernelFunc: C
+  kernelFunc: K
 };
-b(K);
+b(L);

package/dist/ops/fusedSoftmax.js

@@ -1,4 +1,4 @@
-import { e as t } from "../index-C0dhsYom.js";
+import { e as t } from "../index-UdZhlibC.js";
 import "./cpu/fusedSoftmax.js";
 import "./webgl/fusedSoftmax.js";
 import "./grads/fusedSoftmax.js";

package/dist/ops/gatherSub.js

@@ -1,4 +1,4 @@
-import { e as n } from "../index-C0dhsYom.js";
+import { e as n } from "../index-UdZhlibC.js";
 import "./cpu/gatherSub.js";
 import "./webgl/gatherSub.js";
 function f(r, e, t) {

package/dist/ops/gelu.js

@@ -1,7 +1,7 @@
-import "../index-C0dhsYom.js";
+import "../index-UdZhlibC.js";
 import "./cpu/gelu.js";
 import "./webgl/gelu.js";
-import { d as e, g as i } from "../gelu-uyHP1x1f.js";
+import { d as e, g as i } from "../gelu-BFwVnd1r.js";
 export {
   e as dGelu,
   i as gelu

package/dist/ops/grads/attentionMask.js

@@ -1,4 +1,4 @@
-import { i as m, b as i } from "../../index-C0dhsYom.js";
+import { i as m, b as i } from "../../index-UdZhlibC.js";
 import { matMulMul as u } from "../matMulMul.js";
 const p = {
   kernelName: "AttentionMask",

package/dist/ops/grads/fusedSoftmax.js

@@ -1,6 +1,6 @@
-import { i as f, a as i, c as l } from "../../index-C0dhsYom.js";
+import { i as f, a as i, c as l } from "../../index-UdZhlibC.js";
 import { mulDrop as g } from "../mulDrop.js";
-import { s as T } from "../../sum-B8wEpKsg.js";
+import { s as T } from "../../sum-NLYbiDag.js";
 const Y = {
   kernelName: "FusedSoftmax",
   outputsToSave: [!0],

package/dist/ops/grads/gelu.js

@@ -1,5 +1,5 @@
-import "../../index-C0dhsYom.js";
-import { a as e } from "../../gelu-uyHP1x1f.js";
+import "../../index-UdZhlibC.js";
+import { a as e } from "../../gelu-BFwVnd1r.js";
 export {
   e as geluGradConfig
 };

package/dist/ops/grads/matMulGelu.js

@@ -1,4 +1,4 @@
-import { i as a, e as o } from "../../index-C0dhsYom.js";
+import { i as a, e as o } from "../../index-UdZhlibC.js";
 function i(e, n, r) {
   return o().runKernel("MatMulGeluGrad", { dy: e, x: n, kernel: r });
 }

package/dist/ops/grads/normRMS.js

@@ -1,4 +1,4 @@
-import { i as t, e as g } from "../../index-C0dhsYom.js";
+import { i as t, e as g } from "../../index-UdZhlibC.js";
 function i(r, a, n) {
   return g().runKernel("RMSNormGrad", { dy: r, x: a, gamma: n });
 }

package/dist/ops/grads/qkv.js

@@ -1,4 +1,4 @@
-import { i as Q } from "../../index-C0dhsYom.js";
+import { i as Q } from "../../index-UdZhlibC.js";
 const V = {
   kernelName: "QKV",
   inputsToSave: ["x", "kernel"],

package/dist/ops/grads/rope.js

@@ -1,4 +1,4 @@
-import { i, e as a } from "../../index-C0dhsYom.js";
+import { i, e as a } from "../../index-UdZhlibC.js";
 function p(n, e, s, o) {
   return a().runKernel("Rope", { x: n, sin: e, cos: s }, { pastLen: o });
 }

package/dist/ops/matMulGelu.js

@@ -1,4 +1,4 @@
-import { e as u } from "../index-C0dhsYom.js";
+import { e as u } from "../index-UdZhlibC.js";
 import "./cpu/matMulGelu.js";
 import "./webgl/matMulGelu.js";
 import "./grads/matMulGelu.js";

package/dist/ops/matMulMul.js

@@ -1,4 +1,4 @@
-import { e as u } from "../index-C0dhsYom.js";
+import { e as u } from "../index-UdZhlibC.js";
 import "./cpu/matMulMul.js";
 import "./webgl/matMulMul.js";
 function m(e, r, t, l = !1, n = !1) {

package/dist/ops/mulDrop.js

@@ -1,4 +1,4 @@
-import { e as t } from "../index-C0dhsYom.js";
+import { e as t } from "../index-UdZhlibC.js";
 import "./cpu/mulDropout.js";
 import "./webgl/mulDropout.js";
 function m(r, o, e, n) {

package/dist/ops/normRMS.js

@@ -1,4 +1,4 @@
-import { e as n } from "../index-C0dhsYom.js";
+import { e as n } from "../index-UdZhlibC.js";
 import "./cpu/normRMS.js";
 import "./webgl/normRMS.js";
 import "./grads/normRMS.js";

package/dist/ops/qkv.js

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

package/dist/ops/rope.js

@@ -1,8 +1,8 @@
-import { e as p } from "../index-C0dhsYom.js";
-import "../random_width-DkYP8W8N.js";
-import "../register_all_kernels-guvSxp7M.js";
+import { e as p } from "../index-UdZhlibC.js";
+import "../random_width-DbSpgl4o.js";
+import "../register_all_kernels-DnbAyBXt.js";
 import "../index-Tf7vU29b.js";
-import "../dataset-JDyjG3QR.js";
+import "../dataset-U3PrjwgU.js";
 import "./cpu/rope.js";
 import "./webgl/rope.js";
 import "./grads/rope.js";

package/dist/ops/scatterSub.js

@@ -1,4 +1,4 @@
-import { e as i } from "../index-C0dhsYom.js";
+import { e as i } from "../index-UdZhlibC.js";
 import "./cpu/scatterSub.js";
 import "./webgl/scatterSub.js";
 function c(t, r, e) {

package/dist/ops/webgl/adamAdjust.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/ops/webgl/adamAdjust.js

@@ -0,0 +1,50 @@
+import { r as n } from "../../Reshape-CEsEp0AI.js";
+import { f } from "../../index-UdZhlibC.js";
+class v {
+  variableNames = ["moments", "value"];
+  outputShape;
+  userCode;
+  customUniforms = [
+    { name: "invBeta1", type: "float" },
+    { name: "invBeta2", type: "float" },
+    { name: "learningRate", type: "float" },
+    { name: "epsilon", type: "float" }
+  ];
+  constructor(a) {
+    this.outputShape = a, this.userCode = `
+        void main() {
+            float v = getValueAtOutCoords();
+            int coords = getOutputCoords();
+            coords *= 2;
+            float m1 = getMoments(coords);
+            float m2 = getMoments(coords + 1);
+
+            float m1Hat = m1 * invBeta1;
+            float m2Hat = m2 * invBeta2;
+
+            float invSqrt = inversesqrt(max(m2Hat, 1e-30));
+            float invDenom = invSqrt / (1.0 + epsilon * invSqrt);
+            float adjustedValue = -learningRate * m1Hat * invDenom + v;
+
+            setOutput(adjustedValue);
+        }
+        `;
+  }
+}
+function h(t) {
+  const { moments: a, value: s } = t.inputs, { beta1: i, beta2: l, learningRate: u, epsilon: p } = t.attrs, e = t.backend, r = n({ inputs: { x: a }, backend: e, attrs: { shape: [-1] } }), o = n({ inputs: { x: s }, backend: e, attrs: { shape: [-1] } }), d = new v(o.shape), m = e.runWebGLProgram(d, [r, o], "float32", [
+    [1 / i],
+    [1 / l],
+    [u],
+    [p]
+  ]);
+  e.disposeIntermediateTensorInfo(r), e.disposeIntermediateTensorInfo(o);
+  const c = n({ inputs: { x: m }, backend: e, attrs: { shape: s.shape } });
+  return e.disposeIntermediateTensorInfo(m), c;
+}
+const g = {
+  kernelName: "AdamAdjust",
+  backendName: "webgl",
+  kernelFunc: h
+};
+f(g);

package/dist/ops/webgl/adamMoments.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/ops/webgl/adamMoments.js

@@ -0,0 +1,38 @@
+import { f as m } from "../../index-UdZhlibC.js";
+class i {
+  variableNames = ["moments", "gradient"];
+  outputShape;
+  userCode;
+  customUniforms = [
+    { name: "beta1", type: "float" },
+    { name: "beta2", type: "float" }
+  ];
+  constructor(e) {
+    this.outputShape = e;
+    const t = e.length, n = t === 1 ? "int" : `ivec${Math.min(t, 4)}`, a = t === 1 ? "coords" : `coords[${t - 1}]`, s = t === 1 ? "coords" : e.slice(0, -1).map((r, c) => `coords[${c}]`).join(", ");
+    this.userCode = `
+        void main() {
+            float m = getMomentsAtOutCoords();
+            ${n} coords = getOutputCoords();
+            float g = getGradient(${s});
+            int which = ${a};
+
+            float beta = which == 0 ? beta1 : beta2;
+            float gg = which == 0 ? g : g * g;
+    
+            float newM = m * beta + gg * (1.0 - beta);
+            setOutput(newM);
+        }
+        `;
+  }
+}
+function d(o) {
+  const { moments: e, gradient: t } = o.inputs, { beta1: n, beta2: a } = o.attrs, s = o.backend, r = new i(e.shape);
+  return s.runWebGLProgram(r, [e, t], "float32", [[n], [a]]);
+}
+const g = {
+  kernelName: "AdamMoments",
+  backendName: "webgl",
+  kernelFunc: d
+};
+m(g);

package/dist/ops/webgl/appendCache.js

@@ -1,4 +1,4 @@
-import { f as p } from "../../index-C0dhsYom.js";
+import { f as p } from "../../index-UdZhlibC.js";
 class m {
   variableNames = ["cache", "item"];
   outputShape;

package/dist/ops/webgl/attentionMask.js

@@ -1,4 +1,4 @@
-import { f as m } from "../../index-C0dhsYom.js";
+import { f as m } from "../../index-UdZhlibC.js";
 class h {
   variableNames = ["q", "k"];
   outputShape;

package/dist/ops/webgl/fusedSoftmax.js

@@ -1,7 +1,7 @@
-import { m as b, s as I, r as k } from "../../RealDiv-C4hOvYOZ.js";
-import { r as v } from "../../Reshape-BLijOA8h.js";
-import { f as w, p as P } from "../../index-C0dhsYom.js";
-import { e as S } from "../../axis_util-DaAl5MER.js";
+import { m as b, s as I, r as k } from "../../RealDiv-CVYNbZxu.js";
+import { r as v } from "../../Reshape-CEsEp0AI.js";
+import { f as w, p as P } from "../../index-UdZhlibC.js";
+import { e as S } from "../../axis_util-5DTW2tFV.js";
 class T {
   variableNames = ["logits", "maxLogits"];
   outputShape;

package/dist/ops/webgl/gatherSub.js

@@ -1,17 +1,17 @@
-import { f as l } from "../../index-C0dhsYom.js";
+import { f as l } from "../../index-UdZhlibC.js";
 class u {
   variableNames = ["labels", "logits", "values"];
   outputShape;
   userCode;
   constructor(e) {
     this.outputShape = [e], this.userCode = `
-      void main() {
-        int coords = getOutputCoords();
-        int index = int(getLabelsAtOutCoords());
-        float val = getValuesAtOutCoords();
-        float logit = getLogits(coords, index);
-        setOutput(val - logit);
-      }
+          void main() {
+              int coords = getOutputCoords();
+              int index = int(getLabelsAtOutCoords());
+              float val = getValuesAtOutCoords();
+              float logit = getLogits(coords, index);
+              setOutput(val - logit);
+          }
     `;
   }
 }

package/dist/ops/webgl/gelu.js

@@ -1,5 +1,5 @@
-import { f as a } from "../../index-C0dhsYom.js";
-import { u as s, C as x } from "../../kernel_funcs_utils-CwRTFqrc.js";
+import { f as a } from "../../index-UdZhlibC.js";
+import { u as s, C as x } from "../../kernel_funcs_utils-CXDy3EN7.js";
 const t = 0.7978845608028654, r = 0.044715, c = x + `
     float x3 = x * x * x;
     float inner = x + ${r} * x3;

package/dist/ops/webgl/log.js

@@ -1,6 +1,6 @@
-import { f as e, a8 as r } from "../../index-C0dhsYom.js";
-import { u as s, l as N } from "../../kernel_funcs_utils-CwRTFqrc.js";
-import { y as l } from "../../shared-BNa2q6jD.js";
+import { f as e, af as r } from "../../index-UdZhlibC.js";
+import { u as s, l as N } from "../../kernel_funcs_utils-CXDy3EN7.js";
+import { y as l } from "../../shared-DS5waSIY.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/ops/webgl/matMulGelu.js

@@ -1,7 +1,7 @@
-import { f as _, t as R, e as C, j as A, k as N, l as H, u as O } from "../../index-C0dhsYom.js";
-import { r as f } from "../../Reshape-BLijOA8h.js";
-import { M as U } from "../../mulmat_packed_gpu-Gzf3I9UV.js";
-import { m as E } from "../../mat_mul-1nwdPkQ_.js";
+import { f as _, t as R, e as C, j as A, k as N, l as H, u as O } from "../../index-UdZhlibC.js";
+import { r as f } from "../../Reshape-CEsEp0AI.js";
+import { M as U } from "../../mulmat_packed_gpu-IGPBp6h9.js";
+import { m as E } from "../../mat_mul-dwmZz69e.js";
 const M = 0.7978845608028654, x = 0.044715, j = `
     vec4 x3 = x * x * x;
     vec4 inner = x + ${x} * x3;

package/dist/ops/webgl/matMulMul.js

@@ -1,4 +1,4 @@
-import { f as u } from "../../index-C0dhsYom.js";
+import { f as u } from "../../index-UdZhlibC.js";
 import { batchMatMulGeluImpl as c } from "./matMulGelu.js";
 const M = `
     return a * b;

package/dist/ops/webgl/mulDropout.js

@@ -1,4 +1,4 @@
-import { f as m } from "../../index-C0dhsYom.js";
+import { f as m } from "../../index-UdZhlibC.js";
 class f {
   variableNames = ["a", "b"];
   outputShape;

package/dist/ops/webgl/normRMS.js

@@ -1,5 +1,5 @@
-import { f as p, e as G } from "../../index-C0dhsYom.js";
-import { s as x } from "../../sum-B8wEpKsg.js";
+import { f as p, e as G } from "../../index-UdZhlibC.js";
+import { s as x } from "../../sum-NLYbiDag.js";
 class y {
   variableNames = ["x", "meanSquare", "gamma"];
   outputShape;

package/dist/ops/webgl/qkv.js

@@ -1,4 +1,4 @@
-import { f as i } from "../../index-C0dhsYom.js";
+import { f as i } from "../../index-UdZhlibC.js";
 class l {
   variableNames = ["x", "kernel"];
   outputShape;

package/dist/ops/webgl/rope.js

@@ -1,4 +1,4 @@
-import { f as u } from "../../index-C0dhsYom.js";
+import { f as u } from "../../index-UdZhlibC.js";
 class l {
   variableNames = ["x", "sin", "cos"];
   outputShape;

package/dist/ops/webgl/scatterSub.js

@@ -1,4 +1,4 @@
-import { f as i } from "../../index-C0dhsYom.js";
+import { f as i } from "../../index-UdZhlibC.js";
 class u {
   variableNames = ["labels", "softmaxProbs", "dy"];
   outputShape;

package/dist/ops/webgpu/adamAdjust.d.ts

@@ -0,0 +1 @@
+export {};