@genai-fi/nanogpt 0.8.5 → 0.9.0

package/dist/{range-BcUvLuf5.js → range-CWcz7xFA.js}

@@ -1,4 +1,4 @@
-import { E as e, R as f } from "./index-DdmHGZjq.js";
+import { G as e, R as f } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.
@@ -15,12 +15,12 @@ import { E as e, R as f } from "./index-DdmHGZjq.js";
  * limitations under the License.
  * =============================================================================
  */
-function E(n, o, r = 1, a = "float32") {
+function i(n, o, r = 1, a = "float32") {
   if (r === 0)
     throw new Error("Cannot have a step of zero");
   const t = { start: n, stop: o, step: r, dtype: a };
   return e.runKernel(f, {}, t);
 }
 export {
-  E as r
+  i as r
 };

package/dist/{reciprocal-DhDWSKiD.js → reciprocal-C4rNcM-S.js}

@@ -1,4 +1,4 @@
-import { C as c, D as n, E as e, Z as s } from "./index-DdmHGZjq.js";
+import { E as c, F as n, G as e, Z as s } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{register_all_kernels-Do9VvZmo.js → register_all_kernels-DIGpEwcf.js}

@@ -1,36 +1,36 @@
-import { C as ae, D as ne, E as he, co as ps, cp as hs, cr as Zs, cs as fs, ct as ms, cv as gs, Q as Ro, cu as xs, cw as Cs, bw as To, n as Y, df as Js, cC as en, o as Do, L as tn, by as Eo, cG as bs, cH as Is, cI as ys, aq as wc, cL as sn, J as Nc, a as nn, l as Se, dn as $s, cO as on, a0 as It, bC as Fo, cS as ks, di as an, dl as rn, b as Po, du as cn, cU as ws, dx as ln, dy as un, ap as xe, dz as S, aW as vc, q as Sc, T as Oo, dA as Rc, cV as Tc, x as Dc, w as Ec, aX as Fc, c as Pc, ag as V, d4 as Ao, d5 as _o, e as Me, aR as Lt, d9 as Vo, aP as Lo, d6 as Oc, aO as Wt, an as Ke, da as Wo, as as dn, at as Ac, j as z, bn as Ge, au as _c, _ as pn, av as hn, V as Vc, a5 as Lc, U as Mo, a3 as se, cg as Bo, cq as Go, bJ as zo, p as me, ab as st, bK as Uo, bL as Ho, bM as Xo, af as ct, bN as Ko, cx as qo, cy as jo, bQ as Yo, bP as Qo, aS as Zo, de as Jo, bR as fn, cA as ea, a$ as ta, I as sa, bS as na, bE as Te, cc as oa, bT as aa, bU as ra, cB as ia, bV as ca, ac as mn, bW as gn, bd as la, bX as ua, u as nt, N as Wc, bY as da, bZ as pa, b_ as ha, ce as fa, cf as ma, dg as ga, bv as Ns, bp as Ft, bs as xn, cD as Xn, dB as mt, dC as xa, cE as Kn, dh as Mc, a1 as Bc, aU as Ca, cF as ba, b$ as Cn, aY as Ia, aC as ya, aA as yt, db as $a, dc as ka, be as wa, cd as Na, dk as va, a8 as Sa, G as Ra, dm as Ta, b2 as bn, b4 as In, c1 as yn, cK as Da, ah as Gc, c2 as Ea, c3 as Fa, cM as Pa, cN as Oa, dp as Aa, b5 as _a, aV as Va, bo as La, Y as $n, S as Wa, a6 as Ma, bg as Ba, bi as Ga, bh as za, c4 as Ua, cW as Ha, c5 as Xa, P as Ka, a4 as qa, c6 as ja, cP as kn, dD as Ya, dE as Qa, dF as Za, R as Ja, Z as wn, bj as er, cQ as tr, bk as sr, cR as nr, c8 as or, bf as ar, b9 as Nn, a7 as rr, dq as ir, aT as cr, c9 as vn, ad as Sn, ca as Rn, c0 as Tn, c7 as lr, dG as ur, dH as dr, dr as pr, ds as hr, dt as fr, W as mr, cT as gr, aw as Dn, dv as xr, dI as Cr, dJ as br, bO as En, dw as Ir, X as yr, bl as $r, ay as Fn, bb as kr, cb as wr, f as Nr, dK as vr, dL as qn, d7 as jn, d8 as zc, t as Uc, dM as Hc, dN as Xc, bB as Kc, bc as qc, bm as jc, bt as Yc, $ as Qc, bu as Zc, b7 as Jc, b8 as el, bq as tl, br as sl, aL as Sr, bx as nl, ai as ol, ao as al, bz as rl, dj as il, aZ as cl, a_ as ll, b0 as ul, b1 as dl, aj as pl, bA as hl, b6 as fl, bH as ml, b3 as gl, aB as Rr, bD as xl, ar as Cl, bF as bl, ba as Il, bG as yl } from "./index-DdmHGZjq.js";
-import { e as Tr, s as Dr, l as Er, a as Fr, b as Pr, m as Or, c as Ar, r as Ms, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Vl, p as Ll, A as Wl, B as Ml, C as Bl, F as Gl, G as zl, H as Ul, I as Hl } from "./random_width-DKGeiFuR.js";
-import { r as re } from "./reshape-WeJkT3ja.js";
-import { b as Xl } from "./broadcast_to-CUvOdOT5.js";
-import { c as os } from "./concat-pHiVqR3L.js";
-import { c as Kl, s as ql } from "./sin-CPxad7Am.js";
-import { w as _r, r as Vr, n as ss, b as Bs, v as Gs, x as Lr, h as jl, g as Yl, e as Ql, c as Zl, f as Jl, j as eu, m as tu, q as su, k as nu, p as ou, y as au, z as ru, A as iu, B as cu, s as lu, d as uu, t as du, u as pu } from "./ops-542ai2vG.js";
-import { m as Tt } from "./mat_mul-Dpy2mMRu.js";
-import { w as Wr, k as Ye, h as Le, x as hu, y as fu, z as mu, C as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as vs, S as Mr, a as Br, A as Gr, B as zr, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-BgUO9gHY.js";
-import { e as $u, l as ku, a as wu, m as Nu } from "./log_sum_exp-C8yFJfZz.js";
-import { f as vu } from "./dropout-CcKSfOYE.js";
-import { g as Su } from "./gather-CPg6ZlQA.js";
-import { c as zs, z as Ur } from "./zeros-Dnwix0p4.js";
-import { s as as } from "./slice-B7jXtPnp.js";
-import { m as Ru } from "./mod-CbibJi3D.js";
-import { r as Tu } from "./reciprocal-DhDWSKiD.js";
-import { r as Du } from "./relu-B1AXs7p5.js";
-import { s as Us } from "./split-BPxr8_8m.js";
-import { s as Eu } from "./softmax-BfsyI4As.js";
-import { s as Fu } from "./stack-BNwLzE43.js";
-import { s as Pu } from "./sum-ByFINZgi.js";
-import { a as j, u as le, c as ut, i as dt, b as Ou, d as Pt, t as Pe, e as kt, f as gt, g as Hr, r as Ot, h as Be, j as Au, k as _u, l as Pn, z as Vu, m as On, n as Xr, o as Lu, p as Wu, q as Mu, v as Bu, w as Gu, x as zu, y as Uu, A as Hu, B as Xu, C as Ku, D as ft, E as qu, F as ju, G as Kr, H as Yu, I as Qu, J as Zu, K as Ju, L as ed, M as td, N as sd, O as nd, P as od, Q as ad, R as rd, S as id, T as cd, U as ld, V as ud, W as dd, X as pd, Y as hd, Z as fd, _ as md, $ as gd, a0 as xd, a1 as Cd, a2 as bd, a3 as Id, a4 as yd, a5 as $d, a6 as kd, a7 as wd, a8 as Nd, a9 as vd, aa as Sd, ab as Rd, ac as Td, ad as Dd, ae as Ed, af as Fd, ag as Pd, ah as Od, ai as Ad } from "./shared-V6D_md-c.js";
-import { m as xt, A as Ss, B as Rs, C as Ts, D as qr, F as jr, H as Yr, G as je, I as Qr, J as Zr, K as Jr, L as ei, M as ti, z as si, y as ni, x as oi, w as ai, u as ri, E as ii, T as ci, U as _d, V as Vd, W as Ld, X as Wd, Y as Md, Z as Bd, N as li, O as ui, P as di, Q as pi, c as hi, f as Gd, S as zd, _ as Ud } from "./backend_util-yC3YH1jo.js";
-import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Ds } from "./axis_util-Did9235A.js";
+import { E as ae, F as ne, G as he, co as ps, cp as hs, cr as Zs, cs as fs, ct as ms, cv as gs, Q as Ro, cu as xs, cw as Cs, bw as To, n as Y, df as Js, cC as en, o as Do, A as tn, by as Eo, cG as bs, cH as Is, cI as ys, aq as wc, cL as sn, z as Nc, a as nn, l as Se, dn as $s, cO as on, a0 as It, bC as Fo, cS as ks, di as an, dl as rn, b as Po, du as cn, cU as ws, dx as ln, dy as un, ap as xe, dz as S, aW as vc, q as Sc, T as Oo, dA as Rc, cV as Tc, x as Dc, w as Ec, aX as Fc, c as Pc, ag as V, d4 as Ao, d5 as _o, e as Me, aR as Lt, d9 as Vo, aP as Lo, d6 as Oc, aO as Wt, an as Ke, da as Wo, as as dn, at as Ac, j as z, bn as Ge, au as _c, _ as pn, av as hn, V as Vc, a5 as Lc, U as Mo, a3 as se, cg as Bo, cq as Go, bJ as zo, p as me, ab as st, bK as Uo, bL as Ho, bM as Xo, af as ct, bN as Ko, cx as qo, cy as jo, bQ as Yo, bP as Qo, aS as Zo, de as Jo, bR as fn, cA as ea, a$ as ta, K as sa, bS as na, bE as Te, cc as oa, bT as aa, bU as ra, cB as ia, bV as ca, ac as mn, bW as gn, bd as la, bX as ua, u as nt, N as Wc, bY as da, bZ as pa, b_ as ha, ce as fa, cf as ma, dg as ga, bv as Ns, bp as Ft, bs as xn, cD as Xn, dB as mt, dC as xa, cE as Kn, dh as Mc, a1 as Bc, aU as Ca, cF as ba, b$ as Cn, aY as Ia, aC as ya, aA as yt, db as $a, dc as ka, be as wa, cd as Na, dk as va, a8 as Sa, H as Ra, dm as Ta, b2 as bn, b4 as In, c1 as yn, cK as Da, ah as Gc, c2 as Ea, c3 as Fa, cM as Pa, cN as Oa, dp as Aa, b5 as _a, aV as Va, bo as La, Y as $n, S as Wa, a6 as Ma, bg as Ba, bi as Ga, bh as za, c4 as Ua, cW as Ha, c5 as Xa, P as Ka, a4 as qa, c6 as ja, cP as kn, dD as Ya, dE as Qa, dF as Za, R as Ja, Z as wn, bj as er, cQ as tr, bk as sr, cR as nr, c8 as or, bf as ar, b9 as Nn, a7 as rr, dq as ir, aT as cr, c9 as vn, ad as Sn, ca as Rn, c0 as Tn, c7 as lr, dG as ur, dH as dr, dr as pr, ds as hr, dt as fr, W as mr, cT as gr, aw as Dn, dv as xr, dI as Cr, dJ as br, bO as En, dw as Ir, X as yr, bl as $r, ay as Fn, bb as kr, cb as wr, f as Nr, dK as vr, dL as qn, d7 as jn, d8 as zc, t as Uc, dM as Hc, dN as Xc, bB as Kc, bc as qc, bm as jc, bt as Yc, $ as Qc, bu as Zc, b7 as Jc, b8 as el, bq as tl, br as sl, aL as Sr, bx as nl, ai as ol, ao as al, bz as rl, dj as il, aZ as cl, a_ as ll, b0 as ul, b1 as dl, aj as pl, bA as hl, b6 as fl, bH as ml, b3 as gl, aB as Rr, bD as xl, ar as Cl, bF as bl, ba as Il, bG as yl } from "./index-BzFyqcy-.js";
+import { e as Tr, s as Dr, l as Er, a as Fr, b as Pr, m as Or, c as Ar, r as Ms, f as $l, g as kl, h as wl, i as Nl, j as vl, k as Sl, n as Rl, o as Tl, q as Dl, t as El, u as Fl, v as Pl, w as Ol, x as Al, y as _l, z as Vl, p as Ll, A as Wl, B as Ml, C as Bl, F as Gl, G as zl, H as Ul, I as Hl } from "./random_width-CXVRloNK.js";
+import { r as re } from "./reshape-CnIwVG1c.js";
+import { b as Xl } from "./broadcast_to-DzlNweb8.js";
+import { c as os } from "./concat-B912vBbo.js";
+import { c as Kl, s as ql } from "./sin-gpDNRxE0.js";
+import { w as _r, r as Vr, n as ss, b as Bs, v as Gs, x as Lr, h as jl, g as Yl, e as Ql, c as Zl, f as Jl, j as eu, m as tu, q as su, k as nu, p as ou, y as au, z as ru, A as iu, B as cu, s as lu, d as uu, t as du, u as pu } from "./ops-LuCMAnmM.js";
+import { m as Tt } from "./mat_mul-DzjTFx-u.js";
+import { w as Wr, k as Ye, h as Le, x as hu, y as fu, z as mu, C as gu, v as xu, i as ot, j as lt, d as De, e as $t, g as vs, S as Mr, a as Br, A as Gr, B as zr, p as Cu, q as bu, r as Iu, u as yu } from "./selu_util-OtRzVwW5.js";
+import { e as $u, l as ku, a as wu, m as Nu } from "./log_sum_exp-DO6z8tSE.js";
+import { f as vu } from "./dropout-C-csYCLj.js";
+import { g as Su } from "./gather-Dnpgw-YQ.js";
+import { c as zs, z as Ur } from "./zeros-Bj5rMYA7.js";
+import { s as as } from "./slice-d0Vo9XTN.js";
+import { m as Ru } from "./mod-Dobti4j4.js";
+import { r as Tu } from "./reciprocal-C4rNcM-S.js";
+import { r as Du } from "./relu-BjCh_SYb.js";
+import { s as Us } from "./split-DK2k5eHf.js";
+import { s as Eu } from "./softmax-D7Jj3p_P.js";
+import { s as Fu } from "./stack-DFatutCx.js";
+import { s as Pu } from "./sum-CJ0ULhmt.js";
+import { a as j, u as le, c as ut, i as dt, b as Ou, d as Pt, t as Pe, e as kt, f as gt, g as Hr, r as Ot, h as Be, j as Au, k as _u, l as Pn, z as Vu, m as On, n as Xr, o as Lu, p as Wu, q as Mu, v as Bu, w as Gu, x as zu, y as Uu, A as Hu, B as Xu, C as Ku, D as ft, E as qu, F as ju, G as Kr, H as Yu, I as Qu, J as Zu, K as Ju, L as ed, M as td, N as sd, O as nd, P as od, Q as ad, R as rd, S as id, T as cd, U as ld, V as ud, W as dd, X as pd, Y as hd, Z as fd, _ as md, $ as gd, a0 as xd, a1 as Cd, a2 as bd, a3 as Id, a4 as yd, a5 as $d, a6 as kd, a7 as wd, a8 as Nd, a9 as vd, aa as Sd, ab as Rd, ac as Td, ad as Dd, ae as Ed, af as Fd, ag as Pd, ah as Od, ai as Ad } from "./shared-DmRsFyaJ.js";
+import { m as xt, A as Ss, B as Rs, C as Ts, D as qr, F as jr, H as Yr, G as je, I as Qr, J as Zr, K as Jr, L as ei, M as ti, z as si, y as ni, x as oi, w as ai, u as ri, E as ii, T as ci, U as _d, V as Vd, W as Ld, X as Wd, Y as Md, Z as Bd, N as li, O as ui, P as di, Q as pi, c as hi, f as Gd, S as zd, _ as Ud } from "./backend_util-CJIiDoV1.js";
+import { g as ye, a as $e, b as Re, c as Ee, e as Oe, h as Ds } from "./axis_util-TbGYJ208.js";
 import { s as Hd } from "./index-C4L8Cm77.js";
 import { n as fi, b as mi, a as gi } from "./non_max_suppression_impl-CsEgBuMA.js";
-import { c as wt } from "./scatter_nd_util-B7yDhiQr.js";
-import { h as Yn, j as Xd, k as Kd, l as qd, m as jd, n as Yd, o as Qd, P as An, p as ze, u as We, q as xi, c as Ci, T as _e, E as bi, g as Ii, a as yi, r as Zd, s as Jd, t as Z, v as Mt, w as ep, x as Qn, y as tp, z as sp, A as Bt, B as np, C as op, D as Hs, F as jt, G as Yt, H as ap, I as rp, J as Zn, K as ip, L as cp, M as _s, N as lp, O as up, Q as dp, R as Qt, S as Vs, U as pp, f as Ce, V as we, W as Zt, X as Jt, Y as hp, d as Jn, e as eo, i as $i, Z as fp, _ as mp, $ as gp, a0 as xp, a1 as Cp, a2 as bp, a3 as Gt } from "./gpgpu_math-D_ODOLix.js";
-import { s as ki, a as Ip, t as wi, b as yp, c as $p, d as Ni, e as kp, n as wp, f as Np, g as vp, h as Sp, i as Rp, j as Tp, k as Dp, l as Ep, o as Fp, p as Pp, q as Op, r as Ap, u as _p, v as Vp, w as Lp, x as Wp, y as Mp, z as Bp, A as Gp, B as zp, C as Up, D as Hp, E as Xp, F as Kp, G as qp, H as jp, I as Yp, J as Qp, K as Zp, L as Jp, M as vi, N as eh, O as th, P as sh, Q as nh, R as oh, S as ah, T as rh, U as ih, V as ch, W as lh } from "./shared-CZiWmQCI.js";
-import { a as ve, c as uh, U as it, d as tt, e as qe, A as to, f as Nt, B as _n, h as Vn, m as At, u as oe, C as Ue, b as ke, i as Ve, j as Ln, k as pt, l as vt, n as dh, o as ph, p as hh, q as fh } from "./kernel_funcs_utils-CDfFpUab.js";
-import { R as mh, r as H, a as gh } from "./Reshape-Bh_jzKzV.js";
-import { M as Si } from "./mulmat_packed_gpu-q_Gmwyld.js";
-import { t as Ri, s as Wn, a as zt, m as xh, r as Ch, b as bh, c as Ih, d as yh } from "./RealDiv-D_q39E3A.js";
+import { c as wt } from "./scatter_nd_util-BQdz--Gn.js";
+import { h as Yn, j as Xd, k as Kd, l as qd, m as jd, n as Yd, o as Qd, P as An, p as ze, u as We, q as xi, c as Ci, T as _e, E as bi, g as Ii, a as yi, r as Zd, s as Jd, t as Z, v as Mt, w as ep, x as Qn, y as tp, z as sp, A as Bt, B as np, C as op, D as Hs, F as jt, G as Yt, H as ap, I as rp, J as Zn, K as ip, L as cp, M as _s, N as lp, O as up, Q as dp, R as Qt, S as Vs, U as pp, f as Ce, V as we, W as Zt, X as Jt, Y as hp, d as Jn, e as eo, i as $i, Z as fp, _ as mp, $ as gp, a0 as xp, a1 as Cp, a2 as bp, a3 as Gt } from "./gpgpu_math-CDaYiyE_.js";
+import { s as ki, a as Ip, t as wi, b as yp, c as $p, d as Ni, e as kp, n as wp, f as Np, g as vp, h as Sp, i as Rp, j as Tp, k as Dp, l as Ep, o as Fp, p as Pp, q as Op, r as Ap, u as _p, v as Vp, w as Lp, x as Wp, y as Mp, z as Bp, A as Gp, B as zp, C as Up, D as Hp, E as Xp, F as Kp, G as qp, H as jp, I as Yp, J as Qp, K as Zp, L as Jp, M as vi, N as eh, O as th, P as sh, Q as nh, R as oh, S as ah, T as rh, U as ih, V as ch, W as lh } from "./shared-DuP7ue-R.js";
+import { a as ve, c as uh, U as it, d as tt, e as qe, A as to, f as Nt, B as _n, h as Vn, m as At, u as oe, C as Ue, b as ke, i as Ve, j as Ln, k as pt, l as vt, n as dh, o as ph, p as hh, q as fh } from "./kernel_funcs_utils-DKLK0Mg3.js";
+import { R as mh, r as H, a as gh } from "./Reshape-Bowtk9BP.js";
+import { M as Si } from "./mulmat_packed_gpu-BT60jmzP.js";
+import { t as Ri, s as Wn, a as zt, m as xh, r as Ch, b as bh, c as Ih, d as yh } from "./RealDiv-D4EzDsC0.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{relu-B1AXs7p5.js → relu-BjCh_SYb.js}

@@ -1,4 +1,4 @@
-import { C as o, D as e, E as s, _ as t } from "./index-DdmHGZjq.js";
+import { E as o, F as e, G as s, _ as t } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{reshape-WeJkT3ja.js → reshape-CnIwVG1c.js}

@@ -1,4 +1,4 @@
-import { C as o, D as t, E as a, V as p } from "./index-DdmHGZjq.js";
+import { E as o, F as t, G as a, V as p } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{scatter_nd_util-B7yDhiQr.js → scatter_nd_util-BQdz--Gn.js}

@@ -1,4 +1,4 @@
-import { j as p, a3 as w } from "./index-DdmHGZjq.js";
+import { j as p, a3 as w } from "./index-BzFyqcy-.js";
 function k(o, t, r) {
   const n = t.rank > 1 ? t.shape[t.rank - 1] : 1, e = t.rank > 1 ? t.rank - 1 : 1, h = `Must have updates.shape = indices.shape[:batchDim] + shape[sliceDim:], got updates.shape: ${r.shape}, indices.shape: ${t.shape}, shape: ${o}, sliceDim: ${n}, and batchDim: ${e}.`;
   if (r.rank < e)

package/dist/{selu_util-BgUO9gHY.js → selu_util-OtRzVwW5.js}

@@ -1,7 +1,7 @@
-import { n as D, aq as H, C, D as N, E as G, ar as te, as as ne, at as se, au as re, av as ie, aw as oe, a as le, ax as ue, ay as R } from "./index-DdmHGZjq.js";
-import { r as fe } from "./relu-B1AXs7p5.js";
-import { r as ce } from "./reshape-WeJkT3ja.js";
-import { s as ae } from "./sum-ByFINZgi.js";
+import { n as D, aq as H, E as C, F as N, G, ar as te, as as ne, at as se, au as re, av as ie, aw as oe, a as le, ax as ue, ay as O } from "./index-BzFyqcy-.js";
+import { r as fe } from "./relu-BjCh_SYb.js";
+import { r as ce } from "./reshape-CnIwVG1c.js";
+import { s as ae } from "./sum-CJ0ULhmt.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -85,9 +85,9 @@ function he(e, t, n, s, r, u = !1, i = "channelsLast", o) {
     [c, l, a, p, g] = e;
   else
     throw new Error(`Unknown dataFormat ${i}`);
-  const [h, d, A, , m] = t, [b, k, f] = W(n), [w, E, I] = W(s), y = L(h, w), x = L(d, E), $ = L(A, I), { padInfo: _, outDepth: M, outHeight: S, outWidth: F } = me(r, a, p, g, b, k, f, y, x, $, o), V = u ? m * l : m;
+  const [h, d, A, , m] = t, [b, k, f] = W(n), [w, E, I] = W(s), y = L(h, w), x = L(d, E), $ = L(A, I), { padInfo: _, outDepth: M, outHeight: S, outWidth: R } = me(r, a, p, g, b, k, f, y, x, $, o), V = u ? m * l : m;
   let U;
-  return i === "channelsFirst" ? U = [c, V, M, S, F] : i === "channelsLast" && (U = [c, M, S, F, V]), {
+  return i === "channelsFirst" ? U = [c, V, M, S, R] : i === "channelsLast" && (U = [c, M, S, R, V]), {
     batchSize: c,
     dataFormat: i,
     inDepth: a,
@@ -96,7 +96,7 @@ function he(e, t, n, s, r, u = !1, i = "channelsLast", o) {
     inChannels: l,
     outDepth: M,
     outHeight: S,
-    outWidth: F,
+    outWidth: R,
     outChannels: V,
     padInfo: _,
     strideDepth: b,
@@ -118,14 +118,14 @@ function he(e, t, n, s, r, u = !1, i = "channelsLast", o) {
 }
 function ge(e, t, n, s, r) {
   s == null && (s = B(e, t, n));
-  const u = e[0], i = e[1], o = O((u - t + 2 * s) / n + 1, r), c = O((i - t + 2 * s) / n + 1, r);
+  const u = e[0], i = e[1], o = F((u - t + 2 * s) / n + 1, r), c = F((i - t + 2 * s) / n + 1, r);
   return [o, c];
 }
 function pe(e, t, n, s, r, u) {
   r == null && (r = B(e, t[0], s[0]));
   const i = [0, 0, 0, n];
   for (let o = 0; o < 3; o++)
-    e[o] + 2 * r >= t[o] && (i[o] = O((e[o] - t[o] + 2 * r) / s[o] + 1, u));
+    e[o] + 2 * r >= t[o] && (i[o] = F((e[o] - t[o] + 2 * r) / s[o] + 1, u));
   return i;
 }
 function B(e, t, n, s = 1) {
@@ -155,7 +155,7 @@ function de(e, t, n, s, r, u, i, o, c) {
     a = { top: 0, bottom: 0, left: 0, right: 0, type: "VALID" }, p = Math.ceil((t - u + 1) / s), g = Math.ceil((n - i + 1) / r);
   else if (typeof e == "object") {
     const l = c === "channelsLast" ? e[1][0] : e[2][0], h = c === "channelsLast" ? e[1][1] : e[2][1], d = c === "channelsLast" ? e[2][0] : e[3][0], A = c === "channelsLast" ? e[2][1] : e[3][1];
-    a = { top: l, bottom: h, left: d, right: A, type: l === 0 && h === 0 && d === 0 && A === 0 ? "VALID" : "EXPLICIT" }, p = O((t - u + l + h) / s + 1, o), g = O((n - i + d + A) / r + 1, o);
+    a = { top: l, bottom: h, left: d, right: A, type: l === 0 && h === 0 && d === 0 && A === 0 ? "VALID" : "EXPLICIT" }, p = F((t - u + l + h) / s + 1, o), g = F((n - i + d + A) / r + 1, o);
   } else
     throw Error(`Unknown padding parameter: ${e}`);
   return { padInfo: a, outHeight: p, outWidth: g };
@@ -182,7 +182,7 @@ function me(e, t, n, s, r, u, i, o, c, a, p) {
     throw Error(`Unknown padding parameter: ${e}`);
   return { padInfo: g, outDepth: l, outHeight: h, outWidth: d };
 }
-function O(e, t) {
+function F(e, t) {
   if (!t)
     return Math.trunc(e);
   switch (t) {
@@ -498,7 +498,7 @@ function Y(e, t, n, s, r) {
     }
   for (let o = 0; o < u.length; o++) {
     const c = r[o];
-    u[o] < 0 && (u[o] += c), u[o] = R(0, u[o], r[o]);
+    u[o] < 0 && (u[o] += c), u[o] = O(0, u[o], r[o]);
   }
   return u;
 }
@@ -511,16 +511,16 @@ function z(e, t, n, s, r, u) {
   const o = n[r] || 1;
   (e & 1 << r || u & 1 << r || i == null) && (o > 0 ? i = Number.MIN_SAFE_INTEGER : i = Number.MAX_SAFE_INTEGER);
   const c = s[r];
-  return i < 0 && (i += c), i = R(0, i, c - 1), i;
+  return i < 0 && (i += c), i = O(0, i, c - 1), i;
 }
 function ee(e, t, n, s, r, u) {
   let i = t[r];
   const o = n[r] || 1;
   (e & 1 << r || u & 1 << r || i == null) && (o > 0 ? i = Number.MAX_SAFE_INTEGER : i = Number.MIN_SAFE_INTEGER);
   const c = s[r];
-  return i < 0 && (i += c), o > 0 ? i = R(0, i, c) : i = R(-1, i, c - 1), i;
+  return i < 0 && (i += c), o > 0 ? i = O(0, i, c) : i = O(-1, i, c - 1), i;
 }
-function Oe(e, t, n) {
+function Fe(e, t, n) {
   let s = n.length;
   for (let r = 0; r < n.length; r++)
     if (n[r] > 1) {
@@ -532,13 +532,13 @@ function Oe(e, t, n) {
       return !1;
   return !0;
 }
-function Re(e, t) {
+function Oe(e, t) {
   let n = e.length > 0 ? e[e.length - 1] : 1;
   for (let s = 0; s < e.length - 1; s++)
     n += e[s] * t[s];
   return n;
 }
-function Fe(e, t, n) {
+function Re(e, t, n) {
   let s;
   const r = e.shape.length;
   typeof t == "number" ? s = [t, ...new Array(r - 1).fill(0)] : t.length < r ? s = t.concat(new Array(r - t.length).fill(0)) : s = t.slice(), s.forEach((i) => {
@@ -654,12 +654,12 @@ function K(e, t, n, s, r, u) {
 const et = /* @__PURE__ */ Object.freeze(/* @__PURE__ */ Object.defineProperty({
   __proto__: null,
   assertParamsValid: Le,
-  computeFlatOffset: Re,
+  computeFlatOffset: Oe,
   computeOutShape: Ge,
   getNormalizedAxes: Te,
-  isSliceContinous: Oe,
+  isSliceContinous: Fe,
   maskToAxes: Ce,
-  parseSliceParams: Fe,
+  parseSliceParams: Re,
   sliceInfo: Ve,
   startForAxis: z,
   startIndicesWithElidedDims: Q,
@@ -705,12 +705,12 @@ export {
   Qe as m,
   et as n,
   qe as o,
-  Fe as p,
+  Re as p,
   Le as q,
-  Oe as r,
+  Fe as r,
   ze as s,
   P as t,
-  Re as u,
+  Oe as u,
   Ne as v,
   Ee as w,
   $e as x,

package/dist/{shared-V6D_md-c.js → shared-DmRsFyaJ.js}

@@ -1,9 +1,9 @@
-import { n as Lt, bm as Ce, j as V, l as Pt, a3 as K, bn as et, ae as ht, bo as dt, bp as at, aa as De, ab as st, az as We, b8 as $e, bq as Ue, br as ze, bs as Be, bt as At, ap as H, bu as qt, bv as W, bw as _t, bx as Vt, ai as Ct, by as Dt, ao as Wt, bz as $t, aZ as Ut, a_ as zt, b0 as Bt, b1 as jt, aj as Gt, bA as Zt, b6 as Ht, bB as Kt, b3 as je, aA as Ge, b7 as Xt, bc as Ze, bC as He, p as Jt, u as Ke, t as Xe, bD as Qt, bE as It, ar as Yt, $ as Je, bF as te, ba as ee, bG as ne, aP as pt, bH as se, bI as ot } from "./index-DdmHGZjq.js";
-import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-yC3YH1jo.js";
-import { g as pn, a as wn, e as In, c as bn } from "./axis_util-Did9235A.js";
-import { b as xn } from "./broadcast_to-CUvOdOT5.js";
-import { r as En } from "./reshape-WeJkT3ja.js";
-import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-BgUO9gHY.js";
+import { n as Lt, bm as Ce, j as V, l as Pt, a3 as K, bn as et, ae as ht, bo as dt, bp as at, aa as De, ab as st, az as We, b8 as $e, bq as Ue, br as ze, bs as Be, bt as At, ap as H, bu as qt, bv as W, bw as _t, bx as Vt, ai as Ct, by as Dt, ao as Wt, bz as $t, aZ as Ut, a_ as zt, b0 as Bt, b1 as jt, aj as Gt, bA as Zt, b6 as Ht, bB as Kt, b3 as je, aA as Ge, b7 as Xt, bc as Ze, bC as He, p as Jt, u as Ke, t as Xe, bD as Qt, bE as It, ar as Yt, $ as Je, bF as te, ba as ee, bG as ne, aP as pt, bH as se, bI as ot } from "./index-BzFyqcy-.js";
+import { f as it, m as kt, g as Qe, a as Ye, R as tn, v as en, d as nn, e as sn, h as on, i as rn, j as an, k as ln, l as cn, n as un, o as hn, p as fn, q as Nt, r as dn, s as gn, t as mn } from "./backend_util-CJIiDoV1.js";
+import { g as pn, a as wn, e as In, c as bn } from "./axis_util-TbGYJ208.js";
+import { b as xn } from "./broadcast_to-DzlNweb8.js";
+import { r as En } from "./reshape-CnIwVG1c.js";
+import { p as Fn, q as yn, r as kn, u as Nn } from "./selu_util-OtRzVwW5.js";
 import { g as vn } from "./_commonjsHelpers-ByX85dGu.js";
 function Sn(e, t) {
   for (var n = 0; n < t.length; n++) {

package/dist/{shared-CZiWmQCI.js → shared-DuP7ue-R.js}

@@ -1,4 +1,4 @@
-import { s as l } from "./shared-V6D_md-c.js";
+import { s as l } from "./shared-DmRsFyaJ.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sin-CPxad7Am.js → sin-gpDNRxE0.js}

@@ -1,4 +1,4 @@
-import { C as o, D as t, E as c, ac as a, ad as e } from "./index-DdmHGZjq.js";
+import { E as o, F as t, G as c, ac as a, ad as e } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{slice-B7jXtPnp.js → slice-d0Vo9XTN.js}

@@ -1,4 +1,4 @@
-import { C as c, D as e, E as a, $ as l } from "./index-DdmHGZjq.js";
+import { E as c, F as e, G as a, $ as l } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{softmax-BfsyI4As.js → softmax-D7Jj3p_P.js}

@@ -1,4 +1,4 @@
-import { C as r, D as f, E as e, S as i } from "./index-DdmHGZjq.js";
+import { E as r, F as f, G as e, S as i } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{split-BPxr8_8m.js → split-DK2k5eHf.js}

@@ -1,4 +1,4 @@
-import { C as p, D as i, E as a, W as c } from "./index-DdmHGZjq.js";
+import { E as p, F as i, G as a, W as c } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{stack-BNwLzE43.js → stack-DFatutCx.js}

@@ -1,4 +1,4 @@
-import { C as e, F as c, n, E as k, P as i } from "./index-DdmHGZjq.js";
+import { E as e, I as c, n, G as k, P as i } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/dist/{sum-ByFINZgi.js → sum-CJ0ULhmt.js}

@@ -1,4 +1,4 @@
-import { C as e, D as u, a0 as c, E as l, a1 as m } from "./index-DdmHGZjq.js";
+import { E as e, F as u, a0 as c, G as l, a1 as m } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor-DbqgIV9B.js → tensor-CZr4dh61.js}

@@ -1,4 +1,4 @@
-import { A as t, B as a } from "./index-DdmHGZjq.js";
+import { C as t, D as a } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor1d-CtJq5BOv.js → tensor1d-vML0r3q6.js}

@@ -1,4 +1,4 @@
-import { z as o, A as s, B as t } from "./index-DdmHGZjq.js";
+import { B as o, C as s, D as t } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor2d-CObBWBkW.js → tensor2d-D76QGjF3.js}

@@ -1,4 +1,4 @@
-import { z as t, A as s, B as a } from "./index-DdmHGZjq.js";
+import { B as t, C as s, D as a } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{tensor4d-DLtk7Nxh.js → tensor4d-Df1WlVDY.js}

@@ -1,4 +1,4 @@
-import { z as t, A as a, B as s } from "./index-DdmHGZjq.js";
+import { B as t, C as a, D as s } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/training/Adam.js

@@ -1,7 +1,7 @@
 import { adamAdjust as b } from "../ops/adamAdjust.js";
 import { adamMoments as d } from "../ops/adamMoments.js";
-import { O as g, e as h, t as o, d as B } from "../index-DdmHGZjq.js";
-import { z as M } from "../zeros-Dnwix0p4.js";
+import { O as g, e as h, t as o, d as B } from "../index-BzFyqcy-.js";
+import { z as M } from "../zeros-Bj5rMYA7.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/training/AdamExt.js

@@ -1,4 +1,4 @@
-import { a as r, b as c, c as h, e as o } from "../index-DdmHGZjq.js";
+import { a as r, b as c, c as h, e as o } from "../index-BzFyqcy-.js";
 import { AdamOptimizer as g } from "./Adam.js";
 class y extends g {
   constructor(t, e, s, i, a) {

package/dist/training/DatasetBuilder.js

@@ -1,5 +1,5 @@
-import { t as g } from "../index-DdmHGZjq.js";
-import { d as u, i as d } from "../dataset-DPPl-iLT.js";
+import { t as g } from "../index-BzFyqcy-.js";
+import { d as u, i as d } from "../dataset-DlZtKmBq.js";
 import "../index-Tf7vU29b.js";
 /**
  * @license

package/dist/training/FullTrainer.js

@@ -1,6 +1,6 @@
 import y from "./Trainer.js";
 import v from "./Evaluator.js";
-import { d as S } from "../index-DdmHGZjq.js";
+import { d as S } from "../index-BzFyqcy-.js";
 import w from "../utilities/profile.js";
 const f = {
   desiredLoss: 0.01,

package/dist/training/Trainer.js

@@ -1,7 +1,7 @@
 import { DatasetBuilder as m, flattenTokens as c, PAGE_FACTOR as g } from "./DatasetBuilder.js";
 import u from "./AdamExt.js";
-import { t as f, v as y, d as p } from "../index-DdmHGZjq.js";
-import { z as h } from "../zeros-Dnwix0p4.js";
+import { t as f, v as y, d as p } from "../index-BzFyqcy-.js";
+import { z as h } from "../zeros-Bj5rMYA7.js";
 class x {
   constructor(t, e, i = 1e-3) {
     this.tokenizer = e, this.model = t, this.learningRate = i, this.resetOptimizer(), this.datasetBuilder = new m(e, t.config.blockSize);

package/dist/training/sparseCrossEntropy.js

@@ -1,8 +1,8 @@
 import { gatherSub as x } from "../ops/gatherSub.js";
 import { scatterSub as L } from "../ops/scatterSub.js";
-import { J as C, t as u, L as E, c as G } from "../index-DdmHGZjq.js";
-import { s as y } from "../softmax-BfsyI4As.js";
-import { m as z, l as v } from "../log_sum_exp-C8yFJfZz.js";
+import { z as C, t as u, A as E, c as G } from "../index-BzFyqcy-.js";
+import { s as y } from "../softmax-D7Jj3p_P.js";
+import { m as z, l as v } from "../log_sum_exp-DO6z8tSE.js";
 function k(t, s) {
   return u(() => {
     const n = t.shape[t.shape.length - 1], c = t.shape.slice(0, -1).reduce((o, e) => o * e, 1), h = t.shape.length > 2 ? t.reshape([c, n]) : t, p = s.shape.length > 1 ? s.reshape([c]).cast("int32") : s.cast("int32"), r = z(h, -1, !0), a = G(h, r), d = v(a, -1);

package/dist/utilities/dummy.js

@@ -1,5 +1,5 @@
-import { m as y, v as P, e as S } from "../index-DdmHGZjq.js";
-import { z as i } from "../zeros-Dnwix0p4.js";
+import { m as y, v as P, e as S } from "../index-BzFyqcy-.js";
+import { z as i } from "../zeros-Bj5rMYA7.js";
 async function w(s) {
   const t = i([1, s.config.blockSize], "int32"), [e, n] = s.forward({ training: !1 }, t);
   await e.data(), e.dispose(), n && n.dispose(), t.dispose();

package/dist/utilities/multinomialCPU.js

@@ -1,5 +1,5 @@
-import "../index-DdmHGZjq.js";
-import { t as e } from "../tensor2d-CObBWBkW.js";
+import "../index-BzFyqcy-.js";
+import { t as e } from "../tensor2d-D76QGjF3.js";
 function l(n) {
   let r = 0;
   const i = Math.random();

package/dist/utilities/performance.js

@@ -1,4 +1,4 @@
-import { t as s } from "../index-DdmHGZjq.js";
+import { t as s } from "../index-BzFyqcy-.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 { m as a } from "../index-DdmHGZjq.js";
+import { m as a } from "../index-BzFyqcy-.js";
 const s = 1024 * 1024;
 class l {
   log = /* @__PURE__ */ new Map();

package/dist/utilities/safetensors.js

@@ -1,5 +1,5 @@
-import "../index-DdmHGZjq.js";
-import { t as y } from "../tensor-DbqgIV9B.js";
+import "../index-BzFyqcy-.js";
+import { t as y } from "../tensor-CZr4dh61.js";
 function l(t) {
   if (t === "float32") return "F32";
   if (t === "int32") return "I32";

package/dist/utilities/topP.d.ts

@@ -0,0 +1 @@
+export default function topP(probs: number[][], tP: number): number[];

package/dist/utilities/topP.js

@@ -0,0 +1,13 @@
+function c(s, a) {
+  const t = s[0].map((o, r) => ({ prob: o, index: r })).sort((o, r) => r.prob - o.prob);
+  let n = 0;
+  const e = new Array(t.length).fill(0);
+  for (const o of t)
+    if (n += o.prob, e[o.index] = o.prob, n >= a)
+      break;
+  const b = e.reduce((o, r) => o + r, 0);
+  return e.map((o) => o / b);
+}
+export {
+  c as default
+};

package/dist/utilities/weights.js

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

package/dist/{variable-DPFOJyRG.js → variable-Bm2OFwGI.js}

@@ -1,4 +1,4 @@
-import { E as i } from "./index-DdmHGZjq.js";
+import { G as i } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2018 Google LLC. All Rights Reserved.

package/dist/{webgpu_program-Dhk9R5aG.js → webgpu_program-DkQJOJSd.js}

@@ -1,4 +1,4 @@
-import { ae as k, af as z, ag as E, a3 as j, n as A } from "./index-DdmHGZjq.js";
+import { ae as k, af as z, ag as E, a3 as j, n as A } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2019 Google LLC. All Rights Reserved.

package/dist/{webgpu_util-BqGnZg8t.js → webgpu_util-pLEV9tks.js}

@@ -1,4 +1,4 @@
-import { n as u } from "./index-DdmHGZjq.js";
+import { n as u } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2019 Google LLC. All Rights Reserved.

package/dist/{zeros-Dnwix0p4.js → zeros-Bj5rMYA7.js}

@@ -1,4 +1,4 @@
-import { C as m, D as r, a4 as l, E as c, aa as i, M as p, ab as u, j as f } from "./index-DdmHGZjq.js";
+import { E as m, F as r, a4 as l, G as c, aa as i, M as p, ab as u, j as f } from "./index-BzFyqcy-.js";
 /**
  * @license
  * Copyright 2020 Google LLC. All Rights Reserved.

package/package.json

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