@genai-fi/nanogpt 0.6.3 → 0.7.1

package/dist/{kernel_funcs_utils-CUxJCg0g.js → kernel_funcs_utils-CXDy3EN7.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, am as K, a6 as W, an as z, ao as V, ac as N, ap as F, u as S } from "./index-UdZhlibC.js";
+import { u as O, f as Y } from "./gpgpu_math-DffelNS-.js";
+import { f as v } from "./backend_util-C9Ut8n0Q.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, y as p, e as o, A as v } from "../index-UdZhlibC.js";
+import { v as _ } from "../variable-BTBkayv_.js";
 class M {
   parent;
   config;

package/dist/layers/CausalSelfAttention.js

@@ -3,14 +3,14 @@ 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 { w as c, t as C } from "../index-UdZhlibC.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";
-class $ extends O {
+import { d as y } from "../random_width-DbSpgl4o.js";
+import { v as b } from "../variable-BTBkayv_.js";
+import { r as k, d as L } from "../dropout-DPfPgWWe.js";
+import { r as N } from "../reshape-BR0eoLYN.js";
+import { m as R } from "../mat_mul-dwmZz69e.js";
+class W extends O {
   divisor;
   index;
   units;
@@ -92,5 +92,5 @@ class $ extends O {
   }
 }
 export {
-  $ as default
+  W as default
 };

package/dist/layers/MLP.js

@@ -1,10 +1,10 @@
-import { t as l } from "../index-BAzbokzv.js";
+import { t as l } from "../index-UdZhlibC.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-BTBkayv_.js";
+import { r as h, d as f } from "../dropout-DPfPgWWe.js";
+import { r as d } from "../reshape-BR0eoLYN.js";
+import { m as c } from "../mat_mul-dwmZz69e.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-UdZhlibC.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-BTBkayv_.js";
+import { o as m } from "../ones-C8Mfln6-.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, x as h, t as n, w as p } from "../index-UdZhlibC.js";
+import { r as c } from "../reciprocal-CGB48wZB.js";
+import { c as f, s as m } from "../sin-CiBxrDqX.js";
+import { r as a } from "../range-D9CZhVlR.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-DbSpgl4o.js";
+import "../index-UdZhlibC.js";
+import { T as e } from "../TiedEmbedding-ccLBFiZi.js";
 import "./BaseLayer.js";
-import "../variable-wSS22xj5.js";
-import "../gather-DjyCjmOD.js";
+import "../variable-BTBkayv_.js";
+import "../gather-Bbh8DHhM.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-UdZhlibC.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 { a0 as y } from "../index-UdZhlibC.js";
 import { dummyPassAsync as h } from "../utilities/dummy.js";
 async function L(e, a, r, t) {
   const n = {

package/dist/loader/oldZipLoad.js

@@ -1,13 +1,17 @@
 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 { a0 as k } from "../index-UdZhlibC.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-U3PrjwgU.js";
 import "../index-Tf7vU29b.js";
+import "../ops/cpu/adamAdjust.js";
+import "../ops/webgl/adamAdjust.js";
+import "../ops/cpu/adamMoments.js";
+import "../ops/webgl/adamMoments.js";
 import "../papaparse.min-C8l2Kvo1.js";
 import "../ops/cpu/scatterSub.js";
 import "../ops/webgl/scatterSub.js";
@@ -19,8 +23,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-DbSpgl4o.js";
+import "../register_all_kernels-DnbAyBXt.js";
 import "../ops/cpu/rope.js";
 import "../ops/webgl/rope.js";
 import "../ops/grads/rope.js";
@@ -34,13 +38,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-BFwVnd1r.js";
 import "../ops/cpu/normRMS.js";
 import "../ops/webgl/normRMS.js";
 import "../ops/grads/normRMS.js";
 import "../ops/webgl/log.js";
 import { importWeights as O } from "../utilities/weights.js";
-async function ft(o) {
+async function yt(o) {
   const n = /* @__PURE__ */ new Map(), s = await o.file("manifest.json")?.async("string");
   if (!s)
     throw new Error("Manifest file not found in the zip archive");
@@ -72,5 +76,5 @@ async function ft(o) {
   return { model: e, tokeniser: l };
 }
 export {
-  ft as default
+  yt as default
 };

package/dist/{log_sum_exp-YEo2h3gb.js → log_sum_exp-BnmCkHWl.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 { B as r, C as p, E as u, ae as E, af as h, p as S, c as $, o as d } from "./index-UdZhlibC.js";
+import { e as K } from "./axis_util-5DTW2tFV.js";
+import { m as T } from "./max-ByjEGoFx.js";
+import { r as m } from "./reshape-BR0eoLYN.js";
+import { s as _ } from "./sum-NLYbiDag.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 A(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 L = /* @__PURE__ */ r({ logSumExp_: A });
 export {
-  v as a,
+  w as a,
   N as e,
-  M as l
+  L as l
 };

package/dist/main.d.ts

@@ -1,3 +1,7 @@
+import { default as CausalSelfAttention } from './layers/CausalSelfAttention';
+import { default as MLP } from './layers/MLP';
+import { default as TransformerBlock } from './layers/TransformerBlock';
+import { default as RoPECache } from './layers/RoPECache';
 export { default as NanoGPT } from './NanoGPTModel';
 export { default as TeachableLLM } from './TeachableLLM';
 export { default as CharTokeniser } from './tokeniser/CharTokeniser';
@@ -11,3 +15,12 @@ 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';
+export declare const layers: {
+    CausalSelfAttention: typeof CausalSelfAttention;
+    MLP: typeof MLP;
+    TransformerBlock: typeof TransformerBlock;
+    RoPECache: typeof RoPECache;
+};
+export { default as AdamExt } from './training/AdamExt';

package/dist/main.js

@@ -1,11 +1,11 @@
-import { default as R } from "./NanoGPTModel.js";
-import { default as q } from "./TeachableLLM.js";
-import { default as A } from "./tokeniser/CharTokeniser.js";
-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 { default as W } from "./NanoGPTModel.js";
+import { default as Y } from "./TeachableLLM.js";
+import { default as _ } from "./tokeniser/CharTokeniser.js";
+import { default as oo } from "./tokeniser/bpe.js";
+import { default as to } from "./utilities/waitForModel.js";
+import { default as eo } from "./data/textLoader.js";
+import { estimateMemoryUsage as po, estimateParameterCount as ao, estimateResources as fo, estimateTrainingMemoryUsage as so, validateConfig as lo } from "./utilities/parameters.js";
+import "./index-UdZhlibC.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-DbSpgl4o.js";
+import "./register_all_kernels-DnbAyBXt.js";
 import "./index-Tf7vU29b.js";
-import "./dataset-pgqp-YfL.js";
+import "./dataset-U3PrjwgU.js";
 import "./ops/cpu/rope.js";
 import "./ops/webgl/rope.js";
 import "./ops/grads/rope.js";
@@ -33,21 +33,42 @@ 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-BFwVnd1r.js";
 import "./ops/cpu/normRMS.js";
 import "./ops/webgl/normRMS.js";
 import "./ops/grads/normRMS.js";
 import "./ops/webgl/log.js";
+import "./ops/cpu/adamMoments.js";
+import "./ops/webgl/adamMoments.js";
+import "./ops/cpu/adamAdjust.js";
+import "./ops/webgl/adamAdjust.js";
+import { selectBackend as xo } from "./backend.js";
+import { default as co } from "./utilities/performance.js";
+import o from "./layers/CausalSelfAttention.js";
+import r from "./layers/MLP.js";
+import t from "./layers/TransformerBlock.js";
+import m from "./layers/RoPECache.js";
+import { default as ko } from "./training/AdamExt.js";
+const O = {
+  CausalSelfAttention: o,
+  MLP: r,
+  TransformerBlock: t,
+  RoPECache: m
+};
 export {
-  I as BPETokeniser,
-  A as CharTokeniser,
-  R as NanoGPT,
-  q as TeachableLLM,
-  V as estimateMemoryUsage,
-  W as estimateParameterCount,
-  X as estimateResources,
-  Y as estimateTrainingMemoryUsage,
-  Q as loadTextData,
-  Z as validateConfig,
-  K as waitForModel
+  ko as AdamExt,
+  oo as BPETokeniser,
+  _ as CharTokeniser,
+  W as NanoGPT,
+  Y as TeachableLLM,
+  po as estimateMemoryUsage,
+  ao as estimateParameterCount,
+  fo as estimateResources,
+  so as estimateTrainingMemoryUsage,
+  O as layers,
+  eo as loadTextData,
+  co as performanceTest,
+  xo as selectBackend,
+  lo as validateConfig,
+  to as waitForModel
 };

package/dist/{mat_mul-7121rsJk.js → mat_mul-dwmZz69e.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 { B as m, C as s, I as c, E as M, J as p } from "./index-UdZhlibC.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-ByjEGoFx.js}

@@ -1,4 +1,4 @@
-import { o as r, q as e, E as x, M as c } from "./index-BAzbokzv.js";
+import { B as r, C as e, E as x, U as c } from "./index-UdZhlibC.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -15,8 +15,8 @@ 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 };
+function m(n, s = null, o = !1) {
+  const t = { x: e(n, "x", "max") }, a = { reductionIndices: s, keepDims: o };
   return x.runKernel(c, t, a);
 }
 const l = /* @__PURE__ */ r({ max_: m });

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

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