@genai-fi/nanogpt 0.18.0 → 0.19.0

package/dist/utilities/dummy.js

@@ -1,10 +1,10 @@
-import { Y as y, e as I, Z as P } from "../index-DSGwv2Yx.js";
-import { z as c } from "../zeros-Bw0puq_w.js";
+import { b as y, h as I, v as P } from "../index-CUXkjxiT.js";
+import { z as c } from "../zeros-DvZpK8s6.js";
 async function w(s) {
   const t = c([1, s.config.blockSize], "int32"), n = s.forward({ training: !1 }, t);
   await n.data(), n.dispose(), t.dispose();
 }
-async function G(s) {
+async function b(s) {
   const t = y(), n = t.numBytesInGPUAllocated ?? t.numBytesAllocatedInGPU ?? t.numBytes;
   await w(s);
   const a = c([1, s.config.blockSize], "int32"), e = {
@@ -32,12 +32,12 @@ async function G(s) {
   }
   return e;
 }
-function b(s) {
+function v(s) {
   const t = c([1, s.config.blockSize], "int32");
   s.forward({ training: !1 }, t).dispose(), t.dispose();
 }
 export {
-  b as dummyPass,
+  v as dummyPass,
   w as dummyPassAsync,
-  G as dummyPassTrainAsync
+  b as dummyPassTrainAsync
 };

package/dist/utilities/multinomialCPU.js

@@ -1,5 +1,5 @@
-import "../index-DSGwv2Yx.js";
-import { t as e } from "../tensor2d-DTtQ1QcT.js";
+import "../index-CUXkjxiT.js";
+import { t as e } from "../tensor2d-BnXMKScO.js";
 function f(n, i) {
   let r = 0;
   const o = i ?? Math.random();

package/dist/utilities/packed.js

@@ -1,4 +1,4 @@
-import { e as n } from "../index-DSGwv2Yx.js";
+import { h as n } from "../index-CUXkjxiT.js";
 function o() {
   return n().backendName === "webgpu";
 }

package/dist/utilities/performance.js

@@ -1,4 +1,4 @@
-import { t as s } from "../index-DSGwv2Yx.js";
+import { t as s } from "../index-CUXkjxiT.js";
 async function f(e, o = 10, r = !1) {
   for (let t = 0; t < 100; t++) {
     const a = r ? await e() : s(e);

package/dist/utilities/profile.js

@@ -1,4 +1,4 @@
-import { Y as a } from "../index-DSGwv2Yx.js";
+import { b as a } from "../index-CUXkjxiT.js";
 const s = 1024 * 1024;
 class l {
   log = /* @__PURE__ */ new Map();

package/dist/utilities/safetensors.js

@@ -1,5 +1,5 @@
-import "../index-DSGwv2Yx.js";
-import { t as y } from "../tensor-D8e0Gd7c.js";
+import "../index-CUXkjxiT.js";
+import { t as y } from "../tensor-BWFldCso.js";
 function l(t) {
   if (t === "float32") return "F32";
   if (t === "int32") return "I32";

package/dist/utilities/sentences.js

@@ -1,8 +1,8 @@
-import { m as w } from "../index-DSGwv2Yx.js";
-import { t as f } from "../tensor2d-DTtQ1QcT.js";
-import { e as y } from "../expand_dims-D0rBtgT1.js";
-import { s as g } from "../sum-D39FeU5h.js";
-import { c as T } from "../concat-Dqk7Xk7h.js";
+import { m as w } from "../index-CUXkjxiT.js";
+import { t as f } from "../tensor2d-BnXMKScO.js";
+import { e as y } from "../expand_dims-Bkd1YD5x.js";
+import { s as g } from "../sum-BpiwSWvg.js";
+import { c as T } from "../concat-CAQpCret.js";
 const p = 16;
 function A(o, t) {
   if (!t)

package/dist/utilities/weights.js

@@ -1,5 +1,5 @@
-import "../index-DSGwv2Yx.js";
-import { t as p } from "../tensor-D8e0Gd7c.js";
+import "../index-CUXkjxiT.js";
+import { t as p } from "../tensor-BWFldCso.js";
 function h(n) {
   const e = n.reduce((s, o) => s + o.length, 0), a = new Float32Array(e);
   let t = 0;

package/dist/{variable-SSATClyt.js → variable-Ck482e3n.js}

@@ -1,4 +1,4 @@
-import { E as i } from "./index-DSGwv2Yx.js";
+import { E as i } from "./index-CUXkjxiT.js";
 function m(r, a = !0, e, t) {
   return i.makeVariable(r, a, e, t);
 }

package/dist/{webgpu_program-CbjdYLYk.js → webgpu_program-B4HmApL1.js}

@@ -1,4 +1,4 @@
-import { ad as z, ac as F, ab as E, a9 as j, v as A } from "./index-DSGwv2Yx.js";
+import { ad as z, ac as F, ab as E, a9 as j, x as A } from "./index-CUXkjxiT.js";
 function L(t, s) {
   if (Math.max(...t) > 5)
     throw new Error("Cannot symbolically compute strides for rank > 6 tensor.");

package/dist/{webgpu_util-DuofJBMo.js → webgpu_util-DYlGSwOJ.js}

@@ -1,4 +1,4 @@
-import { v as u } from "./index-DSGwv2Yx.js";
+import { x as u } from "./index-CUXkjxiT.js";
 const c = (r) => {
   let t = 1;
   for (let n = 0; n < r.length; n++)
@@ -23,16 +23,16 @@ function p(r, t, n = !1) {
   const a = c(r.x.map((i) => t[i])), o = c(r.y.map((i) => t[i]));
   return a <= 4 ? [4, 16, 1] : o <= 4 ? [16, 4, 1] : [16, 16, 1];
 }
-function M(r, t, n = !1) {
+function x(r, t, n = !1) {
   if (n)
     return [4, 4, 1];
   const a = c(r.x.map((i) => t[i])), o = c(r.y.map((i) => t[i]));
   return a <= 4 ? [1, 2, 1] : o <= 4 ? [2, 1, 1] : [2, 2, 1];
 }
-function h(r) {
+function M(r) {
   return { x: r.map((t, n) => n) };
 }
-function x(r) {
+function h(r) {
   if (r === "float32" || r === "int32" || r === "bool" || r === "string" || r === "packedF16")
     return 4;
   if (r === "complex64")
@@ -52,13 +52,13 @@ var s;
   r[r.MatMulReduceProgram = 0] = "MatMulReduceProgram", r[r.MatMulSplitKProgram = 1] = "MatMulSplitKProgram", r[r.MatMulSmallOutputSizeProgram = 2] = "MatMulSmallOutputSizeProgram", r[r.MatMulPackedProgram = 3] = "MatMulPackedProgram", r[r.MatMulMax = 4] = "MatMulMax";
 })(s || (s = {}));
 export {
-  x as G,
+  h as G,
   s as M,
   d as a,
   b,
   m as c,
   p as d,
-  M as e,
-  h as f,
+  x as e,
+  M as f,
   g as i
 };

package/dist/{zeros-Bw0puq_w.js → zeros-DvZpK8s6.js}

@@ -1,5 +1,5 @@
-import { q as n, L as m, N as i, E as c } from "./index-DSGwv2Yx.js";
-import { c as f } from "./complex-3DpPEG9B.js";
+import { u as n, Q as m, U as i, E as c } from "./index-CUXkjxiT.js";
+import { c as f } from "./complex-DI35Q-gW.js";
 function e(o, r = "float32") {
   if (n(o), r === "complex64") {
     const s = e(o, "float32"), t = e(o, "float32");

package/dist/{zeros_like-rOHr54NY.js → zeros_like-CWjDdwr-.js}

@@ -1,38 +1,38 @@
-import { o as u, n as h, E as d, c7 as st, c8 as rt, c9 as nt, ca as et, cb as it, cc as pt, z as B, cd as at, ce as ht, bY as ut, v as D, cf as ct, cg as ft, d as H, a3 as C, bZ as lt, ch as dt, ci as mt, cj as Dt, aE as yt, ck as It, a2 as wt, m as M, _ as F, cl as xt, cm as gt, J as $, c1 as $t, cn as bt, co as vt, cp as kt, cq as Nt, b as _t, cr as Tt, cs as Et, ct as St, cu as qt, cv as r, a_ as Kt, h as zt, ag as W, cw as Pt, cx as Ct, p as Mt, i as Ot, j as At, a1 as Bt } from "./index-DSGwv2Yx.js";
-import { e as j, a as R, l as X, b as G, d as V, f as Z, m as J, r as E, g as Ht, h as Ft, i as Wt, j as jt, k as Rt, n as Xt, o as Gt, c as Vt, p as Zt, q as Jt, t as Ut, u as Yt, v as Qt, w as Lt, x as to, y as oo, z as so, A as ro, B as no, C as eo, s as io, D as po, E as ao, F as ho } from "./unsorted_segment_sum-BfFVV9Zm.js";
-import { r as l } from "./reshape-BIN71H3p.js";
-import { b as uo } from "./broadcast_to-C_EJTVTZ.js";
-import { c as co } from "./clip_by_value-CLAD4h_I.js";
-import { c as _ } from "./concat-Dqk7Xk7h.js";
-import { w as U, e as fo, l as lo, b as mo, m as Do, a as yo, n as Io } from "./not_equal-CkQKkKZy.js";
-import { m as k } from "./mat_mul-DP86qZtZ.js";
-import { s as Y, k as wo, h as xo, e as go, l as $o, p as bo, r as vo, a as ko } from "./step-DQY6_ABw.js";
-import { n as Q, b as S, c as No, g as _o, l as To, a as Eo, d as So, h as qo, m as Ko, f as zo, e as Po, r as Co, s as Mo, u as Oo } from "./resize_nearest_neighbor-C6_0dAnK.js";
-import { e as Ao } from "./expand_dims-D0rBtgT1.js";
-import { f as Bo } from "./floor-CymuCmTO.js";
-import { g as Ho } from "./gather-DEyjXNb1.js";
-import { c as q } from "./complex-3DpPEG9B.js";
-import { n as N, r as K, i as L, t as Fo } from "./transpose-Qxz-4os3.js";
-import { s as T } from "./slice-BvItlgXu.js";
-import { m as Wo } from "./mod-BXjLYwvM.js";
-import { r as jo } from "./relu-DTvZKBsZ.js";
-import { s as z } from "./split-BN9LkEgS.js";
-import { z as Ro } from "./zeros-Bw0puq_w.js";
-import { s as Xo } from "./softmax-BLGJqdwx.js";
-import { s as Go } from "./squeeze-O_YWJpw_.js";
-import { s as Vo } from "./stack-z6QE7kmP.js";
-import { s as Zo } from "./sum-D39FeU5h.js";
-import { t as Jo } from "./tile-CsWlVKKz.js";
-function Uo(t) {
+import { o as u, q as h, E as d, c7 as st, c8 as rt, c9 as nt, ca as et, cb as it, cc as pt, B, cd as at, ce as ht, bY as ut, x as D, cf as ct, cg as ft, i as H, a3 as C, bZ as lt, ch as dt, ci as mt, cj as Dt, aE as yt, ck as It, a2 as wt, m as M, _ as F, cl as xt, cm as gt, L as $, c1 as $t, cn as bt, co as kt, cp as vt, cq as Nt, a as _t, cr as Tt, cs as Et, ct as St, cu as qt, cv as r, a_ as Kt, j as zt, ag as W, cw as Pt, cx as Ct, p as Mt, l as Ot, n as At, a1 as Bt } from "./index-CUXkjxiT.js";
+import { e as j, a as R, l as X, b as G, d as V, f as Z, m as U, r as E, g as Ht, h as Ft, i as Wt, j as jt, k as Rt, n as Xt, o as Gt, c as Vt, p as Zt, q as Ut, t as Yt, u as Jt, v as Lt, w as Qt, x as to, y as oo, z as so, A as ro, B as no, C as eo, s as io, D as po, E as ao, F as ho } from "./unsorted_segment_sum-DJvk5xnh.js";
+import { r as l } from "./reshape-BDOuCSNW.js";
+import { b as uo } from "./broadcast_to-eS93CCN_.js";
+import { c as co } from "./clip_by_value-DDA7rrcT.js";
+import { c as _ } from "./concat-CAQpCret.js";
+import { w as Y, e as fo, l as lo, b as mo, m as Do, a as yo, n as Io } from "./not_equal-hurPF26l.js";
+import { m as v } from "./mat_mul-DhG0Newp.js";
+import { s as J, k as wo, h as xo, e as go, l as $o, p as bo, r as ko, a as vo } from "./step-BTxPtq1r.js";
+import { n as L, b as S, c as No, g as _o, l as To, a as Eo, d as So, h as qo, m as Ko, f as zo, e as Po, r as Co, s as Mo, u as Oo } from "./resize_nearest_neighbor-BojqlfRe.js";
+import { e as Ao } from "./expand_dims-Bkd1YD5x.js";
+import { f as Bo } from "./floor-BQtb-Azg.js";
+import { g as Ho } from "./gather-qIqEqaGn.js";
+import { c as q } from "./complex-DI35Q-gW.js";
+import { n as N, r as K, i as Q, t as Fo } from "./transpose-9kRxIXWR.js";
+import { s as T } from "./slice-BzS11Qh0.js";
+import { m as Wo } from "./mod-CSdCpRjf.js";
+import { r as jo } from "./relu-J_X6MUzx.js";
+import { s as z } from "./split-C2Sj255c.js";
+import { z as Ro } from "./zeros-DvZpK8s6.js";
+import { s as Xo } from "./softmax-D4q1LJN7.js";
+import { s as Go } from "./squeeze-ho4wLUek.js";
+import { s as Vo } from "./stack-DudVrtmG.js";
+import { s as Zo } from "./sum-BpiwSWvg.js";
+import { t as Uo } from "./tile-CFy-xTO6.js";
+function Yo(t) {
   const s = { x: h(t, "x", "acos") };
   return d.runKernel(st, s);
 }
-const Yo = /* @__PURE__ */ u({ acos_: Uo });
-function Qo(t) {
+const Jo = /* @__PURE__ */ u({ acos_: Yo });
+function Lo(t) {
   const s = { x: h(t, "x", "acosh") };
   return d.runKernel(rt, s);
 }
-const Lo = /* @__PURE__ */ u({ acosh_: Qo });
+const Qo = /* @__PURE__ */ u({ acosh_: Lo });
 function ts(t, o = 0) {
   const n = { x: h(t, "x", "argMin") }, e = { axis: o };
   return d.runKernel(nt, n, e);
@@ -94,7 +94,7 @@ function Is(t, o) {
   let s = h(t, "a", "div"), n = h(o, "b", "div");
   [s, n] = B(s, n);
   const e = H(s, n), i = C(e), p = j(n, i);
-  return U(p, i, e);
+  return Y(p, i, e);
 }
 const ws = /* @__PURE__ */ u({ divNoNan_: Is });
 function xs(t, o) {
@@ -102,29 +102,29 @@ function xs(t, o) {
   D((s.rank === 1 || s.rank === 2) && (n.rank === 1 || n.rank === 2), () => `Error in dot: inputs must all be rank 1 or 2, but got ranks ${s.rank} and ${n.rank}.`);
   const e = s.rank === 1 ? s.size : s.shape[1], i = n.rank === 1 ? n.size : n.shape[0];
   if (D(e === i, () => `Error in dot: inner dimensions of inputs must match, but got ${e} and ${i}.`), s.rank === 1 && n.rank === 1) {
-    const p = l(s, [1, -1]), a = l(n, [-1, 1]), f = k(p, a);
+    const p = l(s, [1, -1]), a = l(n, [-1, 1]), f = v(p, a);
     return l(f, []);
   } else if (s.rank === 1 && n.rank === 2) {
-    const p = l(s, [1, -1]), a = l(n, [n.shape[0], n.shape[1]]), f = k(p, a);
+    const p = l(s, [1, -1]), a = l(n, [n.shape[0], n.shape[1]]), f = v(p, a);
     return l(f, [f.size]);
   } else if (s.rank === 2 && n.rank === 1) {
-    const p = l(n, [-1, 1]), a = k(s, p);
+    const p = l(n, [-1, 1]), a = v(s, p);
     return l(a, [a.size]);
   } else {
     const p = l(n, [n.shape[0], n.shape[1]]);
-    return k(s, p);
+    return v(s, p);
   }
 }
 const gs = /* @__PURE__ */ u({ dot_: xs });
 function $s(t, o = null, s = !1) {
-  return Q(t, "euclidean", o, s);
+  return L(t, "euclidean", o, s);
 }
 const bs = /* @__PURE__ */ u({ euclideanNorm_: $s });
-function vs(t) {
+function ks(t) {
   const s = { x: h(t, "x", "expm1") };
   return d.runKernel(lt, s);
 }
-const ks = /* @__PURE__ */ u({ expm1_: vs });
+const vs = /* @__PURE__ */ u({ expm1_: ks });
 function Ns(t) {
   const s = { x: h(t, "x", "isFinite") };
   return d.runKernel(dt, s);
@@ -152,7 +152,7 @@ function Ks(t, o = 5, s = 1, n = 1, e = 0.5) {
 const zs = /* @__PURE__ */ u({ localResponseNormalization_: Ks });
 function Ps(t) {
   const o = h(t, "x", "logSigmoid");
-  return wt((n) => ({ value: N(R(N(n))), gradFunc: (p) => M(p, Y(N(n))) }))(o);
+  return wt((n) => ({ value: N(R(N(n))), gradFunc: (p) => M(p, J(N(n))) }))(o);
 }
 const Cs = /* @__PURE__ */ u({ logSigmoid_: Ps });
 function Ms(t, o) {
@@ -188,8 +188,8 @@ function Fs(t, o, s, n, e, i, p) {
   const c = xo(f.shape, o, i, e, n), m = [c.dilationHeight, c.dilationWidth];
   let I;
   n === "same" ? I = js([c.filterHeight, c.filterWidth], m) : I = [[0, 0], [0, 0]];
-  const w = m[0] === 1 && m[1] === 1, [x, g] = Ws([c.inHeight, c.inWidth], m, I), b = w ? n : "valid", O = w ? f : G(f, m, x), A = (s === "avg" ? () => Z(O, o, i, b, p) : () => J(O, o, i, b, p))(), v = w ? A : V(A, m, g);
-  return y ? l(v, [v.shape[1], v.shape[2], v.shape[3]]) : v;
+  const w = m[0] === 1 && m[1] === 1, [x, g] = Ws([c.inHeight, c.inWidth], m, I), b = w ? n : "valid", O = w ? f : G(f, m, x), A = (s === "avg" ? () => Z(O, o, i, b, p) : () => U(O, o, i, b, p))(), k = w ? A : V(A, m, g);
+  return y ? l(k, [k.shape[1], k.shape[2], k.shape[3]]) : k;
 }
 function Ws(t, o, s) {
   const n = s.map((c) => c[0]), e = s.map((c) => c[1]), i = t.concat(n, e), p = o.map((c, m) => (c - i[m] % c) % c), a = e.map((c, m) => c + p[m]), f = o.map((c, m) => [n[m], a[m]]), y = o.map((c, m) => [0, p[m]]);
@@ -212,31 +212,31 @@ function Vs(t) {
   return d.runKernel(bt, s);
 }
 const Zs = /* @__PURE__ */ u({ reciprocal_: Vs });
-function Js(t) {
+function Us(t) {
   const s = { x: h(t, "x", "sign") };
-  return d.runKernel(vt, s);
+  return d.runKernel(kt, s);
 }
-const Us = /* @__PURE__ */ u({ sign_: Js });
-function Ys(t) {
+const Ys = /* @__PURE__ */ u({ sign_: Us });
+function Js(t) {
   D(t.dtype === "complex64", () => `The dtype for tf.spectral.fft() must be complex64 but got ${t.dtype}.`);
   const o = { input: t };
-  return d.runKernel(kt, o);
+  return d.runKernel(vt, o);
 }
-const ot = /* @__PURE__ */ u({ fft_: Ys });
-function Qs(t) {
+const ot = /* @__PURE__ */ u({ fft_: Js });
+function Ls(t) {
   D(t.dtype === "complex64", () => `The dtype for tf.spectral.ifft() must be complex64 but got ${t.dtype}.`);
   const o = { input: t };
   return d.runKernel(Nt, o);
 }
-const P = /* @__PURE__ */ u({ ifft_: Qs });
-function Ls(t) {
+const P = /* @__PURE__ */ u({ ifft_: Ls });
+function Qs(t) {
   const o = t.shape[t.shape.length - 1], s = t.size / o;
   let n;
   if (o <= 2) {
     const e = l(t, [s, o]);
     n = P(e);
   } else {
-    const e = [s, 2 * (o - 1)], i = l(K(t), [s, o]), p = l(L(t), [s, o]), a = E(T(i, [0, 1], [s, o - 2]), 1), f = M(E(T(p, [0, 1], [s, o - 2]), 1), _t(-1)), y = _([i, a], 1), c = _([p, f], 1), m = l(q(y, c), [e[0], e[1]]);
+    const e = [s, 2 * (o - 1)], i = l(K(t), [s, o]), p = l(Q(t), [s, o]), a = E(T(i, [0, 1], [s, o - 2]), 1), f = M(E(T(p, [0, 1], [s, o - 2]), 1), _t(-1)), y = _([i, a], 1), c = _([p, f], 1), m = l(q(y, c), [e[0], e[1]]);
     n = P(m);
   }
   if (n = K(n), t.rank === 3 && t.shape[0] !== 0) {
@@ -245,7 +245,7 @@ function Ls(t) {
   }
   return n;
 }
-const tr = /* @__PURE__ */ u({ irfft_: Ls });
+const tr = /* @__PURE__ */ u({ irfft_: Qs });
 function or(t, o) {
   D(t.dtype === "float32", () => `The dtype for rfft() must be real value but got ${t.dtype}`);
   let s = t.shape[t.shape.length - 1];
@@ -259,7 +259,7 @@ function or(t, o) {
     x[t.shape.length - 1] = o - s, e = _([t, Ro(x)], t.shape.length - 1), s = o;
   } else
     e = t;
-  const i = C(e), p = l(q(e, i), [n, s]), a = ot(p), f = Math.floor(s / 2) + 1, y = K(a), c = L(a), m = z(y, [f, s - f], y.shape.length - 1), I = z(c, [f, s - f], c.shape.length - 1), w = e.shape.slice();
+  const i = C(e), p = l(q(e, i), [n, s]), a = ot(p), f = Math.floor(s / 2) + 1, y = K(a), c = Q(a), m = z(y, [f, s - f], y.shape.length - 1), I = z(c, [f, s - f], c.shape.length - 1), w = e.shape.slice();
   return w[e.shape.length - 1] = f, l(q(m[0], I[0]), w);
 }
 const sr = /* @__PURE__ */ u({ rfft_: or });
@@ -306,10 +306,10 @@ r().prototype.abs = function() {
   return this.throwIfDisposed(), Kt(this);
 };
 r().prototype.acos = function() {
-  return this.throwIfDisposed(), Yo(this);
+  return this.throwIfDisposed(), Jo(this);
 };
 r().prototype.acosh = function() {
-  return this.throwIfDisposed(), Lo(this);
+  return this.throwIfDisposed(), Qo(this);
 };
 r().prototype.add = function(t) {
   return this.throwIfDisposed(), zt(this, t);
@@ -402,16 +402,16 @@ r().prototype.cosh = function() {
   return this.throwIfDisposed(), Zt(this);
 };
 r().prototype.cumprod = function(t, o, s) {
-  return this.throwIfDisposed(), Jt(this, t, o, s);
+  return this.throwIfDisposed(), Ut(this, t, o, s);
 };
 r().prototype.cumsum = function(t, o, s) {
-  return this.throwIfDisposed(), Ut(this, t, o, s);
+  return this.throwIfDisposed(), Yt(this, t, o, s);
 };
 r().prototype.depthToSpace = function(t, o) {
   return this.throwIfDisposed(), ms(this, t, o);
 };
 r().prototype.depthwiseConv2d = function(t, o, s, n, e, i) {
-  return this.throwIfDisposed(), Yt(this, t, o, s, n, e, i);
+  return this.throwIfDisposed(), Jt(this, t, o, s, n, e, i);
 };
 r().prototype.dilation2d = function(t, o, s, n, e) {
   return this.throwIfDisposed(), ys(this, t, o, s, n, e);
@@ -432,7 +432,7 @@ r().prototype.equal = function(t) {
   return this.throwIfDisposed(), j(this, t);
 };
 r().prototype.erf = function() {
-  return this.throwIfDisposed(), Qt(this);
+  return this.throwIfDisposed(), Lt(this);
 };
 r().prototype.euclideanNorm = function(t, o) {
   return this.throwIfDisposed(), bs(this, t, o);
@@ -444,7 +444,7 @@ r().prototype.expandDims = function(t) {
   return this.throwIfDisposed(), Ao(this, t);
 };
 r().prototype.expm1 = function() {
-  return this.throwIfDisposed(), ks(this);
+  return this.throwIfDisposed(), vs(this);
 };
 r().prototype.fft = function() {
   return this.throwIfDisposed(), ot(this);
@@ -498,7 +498,7 @@ r().prototype.logSigmoid = function() {
   return this.throwIfDisposed(), Cs(this);
 };
 r().prototype.logSoftmax = function(t) {
-  return this.throwIfDisposed(), Lt(this, t);
+  return this.throwIfDisposed(), Qt(this, t);
 };
 r().prototype.logSumExp = function(t, o) {
   return this.throwIfDisposed(), lo(this, t, o);
@@ -522,10 +522,10 @@ r().prototype.logicalXor = function(t) {
   return this.throwIfDisposed(), As(this, t);
 };
 r().prototype.matMul = function(t, o, s) {
-  return this.throwIfDisposed(), k(this, t, o, s);
+  return this.throwIfDisposed(), v(this, t, o, s);
 };
 r().prototype.maxPool = function(t, o, s, n) {
-  return this.throwIfDisposed(), J(this, t, o, s, n);
+  return this.throwIfDisposed(), U(this, t, o, s, n);
 };
 r().prototype.max = function(t, o) {
   return this.throwIfDisposed(), Do(this, t, o);
@@ -555,7 +555,7 @@ r().prototype.neg = function() {
   return this.throwIfDisposed(), N(this);
 };
 r().prototype.norm = function(t, o, s) {
-  return this.throwIfDisposed(), Q(this, t, o, s);
+  return this.throwIfDisposed(), L(this, t, o, s);
 };
 r().prototype.notEqual = function(t) {
   return this.throwIfDisposed(), Io(this, t);
@@ -588,7 +588,7 @@ r().prototype.relu = function() {
   return this.throwIfDisposed(), jo(this);
 };
 r().prototype.relu6 = function() {
-  return this.throwIfDisposed(), vo(this);
+  return this.throwIfDisposed(), ko(this);
 };
 r().prototype.reshapeAs = function(t) {
   return this.throwIfDisposed(), l(this, t.shape);
@@ -621,10 +621,10 @@ r().prototype.separableConv2d = function(t, o, s, n, e, i) {
   return this.throwIfDisposed(), eo(this, t, o, s, n, e, i);
 };
 r().prototype.sigmoid = function() {
-  return this.throwIfDisposed(), Y(this);
+  return this.throwIfDisposed(), J(this);
 };
 r().prototype.sign = function() {
-  return this.throwIfDisposed(), Us(this);
+  return this.throwIfDisposed(), Ys(this);
 };
 r().prototype.sin = function() {
   return this.throwIfDisposed(), io(this);
@@ -665,7 +665,7 @@ r().prototype.stack = function(t, o) {
   return Vo(s, o);
 };
 r().prototype.step = function(t) {
-  return this.throwIfDisposed(), ko(this, t);
+  return this.throwIfDisposed(), vo(this, t);
 };
 r().prototype.stridedSlice = function(t, o, s, n, e, i, p, a) {
   return this.throwIfDisposed(), nr(this, t, o, s, n, e, i, p, a);
@@ -683,7 +683,7 @@ r().prototype.tanh = function() {
   return this.throwIfDisposed(), ao(this);
 };
 r().prototype.tile = function(t) {
-  return this.throwIfDisposed(), Jo(this, t);
+  return this.throwIfDisposed(), Uo(this, t);
 };
 r().prototype.toBool = function() {
   return this.throwIfDisposed(), $(this, "bool");
@@ -710,7 +710,7 @@ r().prototype.unstack = function(t) {
   return this.throwIfDisposed(), Oo(this, t);
 };
 r().prototype.where = function(t, o) {
-  return this.throwIfDisposed(), U(t, this, o);
+  return this.throwIfDisposed(), Y(t, this, o);
 };
 r().prototype.zerosLike = function() {
   return this.throwIfDisposed(), C(this);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@genai-fi/nanogpt",
-    "version": "0.18.0",
+    "version": "0.19.0",
     "type": "module",
     "main": "dist/main.js",
     "types": "dist/main.d.ts",

package/dist/complex-3DpPEG9B.js

@@ -1,11 +0,0 @@
-import { o as t, n as s, a6 as c, E as m, a7 as r } from "./index-DSGwv2Yx.js";
-function l(o, n) {
-  const a = s(o, "real", "complex"), e = s(n, "imag", "complex");
-  c(a.shape, e.shape, `real and imag shapes, ${a.shape} and ${e.shape}, must match in call to tf.complex().`);
-  const p = { real: a, imag: e };
-  return m.runKernel(r, p);
-}
-const i = /* @__PURE__ */ t({ complex_: l });
-export {
-  i as c
-};

package/dist/mat_mul-DP86qZtZ.js

@@ -1,11 +0,0 @@
-import { o as m, n as s, z as c, E as M, B as p } from "./index-DSGwv2Yx.js";
-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: o, transposeB: l };
-  return M.runKernel(p, r, u);
-}
-const i = /* @__PURE__ */ m({ matMul_: f });
-export {
-  i as m
-};

package/dist/mod-BXjLYwvM.js

@@ -1,11 +0,0 @@
-import { o as r, n, z as m, E as c, M as d } from "./index-DSGwv2Yx.js";
-function p(s, t) {
-  let o = n(s, "a", "mod"), a = n(t, "b", "mod");
-  [o, a] = m(o, a);
-  const e = { a: o, b: a };
-  return c.runKernel(d, e);
-}
-const b = /* @__PURE__ */ r({ mod_: p });
-export {
-  b as m
-};

package/dist/readers-17HLdxVM.js

@@ -1,12 +0,0 @@
-import { d as o, i as n } from "./dataset-DlqAN81i.js";
-import "./index-DSGwv2Yx.js";
-import "./index-Cp39cXWe.js";
-function m(t) {
-  return o(async () => {
-    const r = await t();
-    return n(() => r.next());
-  });
-}
-export {
-  m as g
-};

package/dist/relu-DTvZKBsZ.js

@@ -1,9 +0,0 @@
-import { o, n as e, E as s, A as t } from "./index-DSGwv2Yx.js";
-function u(r) {
-  const n = { x: e(r, "x", "relu") };
-  return s.runKernel(t, n);
-}
-const l = /* @__PURE__ */ o({ relu_: u });
-export {
-  l as r
-};

package/dist/slice-BvItlgXu.js

@@ -1,12 +0,0 @@
-import { o as c, n as e, E as a, S as l } from "./index-DSGwv2Yx.js";
-function p(n, r, o) {
-  const s = e(n, "x", "slice", "string_or_numeric");
-  if (s.rank === 0)
-    throw new Error("Slicing scalar is not possible");
-  const t = { x: s }, i = { begin: r, size: o };
-  return a.runKernel(l, t, i);
-}
-const x = /* @__PURE__ */ c({ slice_: p });
-export {
-  x as s
-};

package/dist/split-BN9LkEgS.js

@@ -1,9 +0,0 @@
-import { o as p, n as i, E as a, H as c } from "./index-DSGwv2Yx.js";
-function e(t, s, n = 0) {
-  const o = { x: i(t, "x", "split") }, r = { numOrSizeSplits: s, axis: n };
-  return a.runKernel(c, o, r);
-}
-const u = /* @__PURE__ */ p({ split_: e });
-export {
-  u as s
-};

package/dist/training/SFTDatasetBuilder.d.ts

@@ -1,23 +0,0 @@
-import { Tensor } from '@tensorflow/tfjs-core';
-import { Conversation, ITokeniser } from '../tokeniser/type';
-import { Dataset } from '@tensorflow/tfjs-data';
-import { Task } from './tasks/Task';
-export declare function buildSFTExample(conversation: Conversation[], ignoreIndex: number, tokenizer: ITokeniser, blockSize: number): {
-    xs: Int32Array;
-    ys: Int32Array;
-} | null;
-export declare class SFTDatasetBuilder {
-    tokenizer: ITokeniser;
-    blockSize: number;
-    constructor(tokenizer: ITokeniser, blockSize?: number);
-    /**
-     * Create SFT dataset from structured conversations.
-     * - Always starts from the beginning of each conversation.
-     * - Pads with eosToken and masks padding.
-     * - Masks non-assistant tokens in labels with ignoreIndex (default -100).
-     */
-    createSFTDataset(conversations: Task[], batchSize?: number, ignoreIndex?: number): Promise<Dataset<{
-        xs: Tensor;
-        ys: Tensor;
-    }>>;
-}