@genai-fi/nanogpt 0.7.3 → 0.8.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (197) hide show
  1. package/dist/Generator.d.ts +25 -2
  2. package/dist/Generator.js +152 -49
  3. package/dist/{RealDiv-Dy0p8Bvo.js → RealDiv-D_q39E3A.js} +13 -13
  4. package/dist/{Reshape-DvudQDvJ.js → Reshape-41YpQqEo.js} +1 -1
  5. package/dist/{Reshape-DH5srBP0.js → Reshape-Bh_jzKzV.js} +5 -5
  6. package/dist/TeachableLLM.d.ts +6 -6
  7. package/dist/TeachableLLM.js +33 -31
  8. package/dist/Trainer.d.ts +13 -2
  9. package/dist/Trainer.js +21 -12
  10. package/dist/{axis_util-BzbKo31C.js → axis_util-Did9235A.js} +3 -3
  11. package/dist/backend.js +2 -2
  12. package/dist/{backend_util-TE7aTPhZ.js → backend_util-yC3YH1jo.js} +58 -58
  13. package/dist/{broadcast_to-CdbwV-Dj.js → broadcast_to-CUvOdOT5.js} +2 -2
  14. package/dist/checks/appendCache.d.ts +1 -0
  15. package/dist/checks/appendCache.js +22 -0
  16. package/dist/checks/attentionMask.d.ts +1 -0
  17. package/dist/checks/attentionMask.js +37 -0
  18. package/dist/checks/check.d.ts +9 -0
  19. package/dist/checks/check.js +20 -0
  20. package/dist/checks/gelu.d.ts +1 -0
  21. package/dist/checks/gelu.js +18 -0
  22. package/dist/checks/index.d.ts +19 -0
  23. package/dist/checks/index.js +21 -0
  24. package/dist/checks/normRMS.d.ts +1 -0
  25. package/dist/checks/normRMS.js +16 -0
  26. package/dist/checks/normRMSGrad.d.ts +1 -0
  27. package/dist/checks/normRMSGrad.js +12 -0
  28. package/dist/checks/qkv.d.ts +1 -0
  29. package/dist/checks/qkv.js +25 -0
  30. package/dist/checks/rope.d.ts +1 -0
  31. package/dist/checks/rope.js +21 -0
  32. package/dist/{concat-CsxrgovM.js → concat-pHiVqR3L.js} +1 -1
  33. package/dist/{dataset-CtdBYwjo.js → dataset-DPPl-iLT.js} +9 -9
  34. package/dist/{dropout-DYs5QFGQ.js → dropout-CcKSfOYE.js} +18 -18
  35. package/dist/exports_initializers-DKk7-bsx.js +16 -0
  36. package/dist/{gather-CMMy2KEG.js → gather-CPg6ZlQA.js} +1 -1
  37. package/dist/{gelu-C-dPj6Ku.js → gelu-BkcmEEyD.js} +1 -1
  38. package/dist/{gpgpu_math-DGNLNL4I.js → gpgpu_math-D_ODOLix.js} +26 -26
  39. package/dist/{index-BoWRt-10.js → index-DdmHGZjq.js} +659 -650
  40. package/dist/{index-CLthM0TO.js → index-evZ57wr4.js} +185 -185
  41. package/dist/{kernel_funcs_utils-BYKWV8Aa.js → kernel_funcs_utils-CDfFpUab.js} +21 -21
  42. package/dist/layers/BaseLayer.d.ts +8 -13
  43. package/dist/layers/BaseLayer.js +25 -13
  44. package/dist/layers/CausalSelfAttention.d.ts +3 -2
  45. package/dist/layers/CausalSelfAttention.js +28 -28
  46. package/dist/layers/MLP.d.ts +3 -2
  47. package/dist/layers/MLP.js +16 -20
  48. package/dist/layers/PositionEmbedding.d.ts +9 -0
  49. package/dist/layers/PositionEmbedding.js +45 -0
  50. package/dist/layers/RMSNorm.d.ts +3 -2
  51. package/dist/layers/RMSNorm.js +6 -6
  52. package/dist/layers/RoPECache.d.ts +1 -1
  53. package/dist/layers/RoPECache.js +4 -4
  54. package/dist/layers/TiedEmbedding.d.ts +3 -2
  55. package/dist/layers/TiedEmbedding.js +29 -7
  56. package/dist/layers/TransformerBlock.d.ts +3 -2
  57. package/dist/layers/TransformerBlock.js +1 -1
  58. package/dist/loader/load.d.ts +2 -2
  59. package/dist/loader/loadHF.d.ts +2 -2
  60. package/dist/loader/loadTransformers.d.ts +4 -2
  61. package/dist/loader/loadTransformers.js +10 -9
  62. package/dist/loader/newZipLoad.d.ts +2 -2
  63. package/dist/loader/oldZipLoad.d.ts +2 -2
  64. package/dist/loader/oldZipLoad.js +44 -51
  65. package/dist/loader/save.d.ts +8 -0
  66. package/dist/loader/save.js +62 -0
  67. package/dist/{log_sum_exp-DbjkV734.js → log_sum_exp-C8yFJfZz.js} +45 -24
  68. package/dist/main.d.ts +6 -4
  69. package/dist/main.js +24 -18
  70. package/dist/{mat_mul-8m8pfdcx.js → mat_mul-Dpy2mMRu.js} +1 -1
  71. package/dist/mod-CbibJi3D.js +27 -0
  72. package/dist/models/NanoGPTV1.d.ts +15 -0
  73. package/dist/models/NanoGPTV1.js +71 -0
  74. package/dist/{config.d.ts → models/config.d.ts} +1 -0
  75. package/dist/{config.js → models/config.js} +1 -0
  76. package/dist/models/factory.d.ts +3 -0
  77. package/dist/models/factory.js +14 -0
  78. package/dist/models/model.d.ts +26 -0
  79. package/dist/models/model.js +70 -0
  80. package/dist/{mulmat_packed_gpu-VSekgsNv.js → mulmat_packed_gpu-q_Gmwyld.js} +1 -1
  81. package/dist/{ones-Dj0SDhHf.js → ones-BAqVh-eA.js} +2 -2
  82. package/dist/ops/adamAdjust.js +1 -1
  83. package/dist/ops/adamMoments.js +1 -1
  84. package/dist/ops/appendCache.js +3 -3
  85. package/dist/ops/attentionMask.js +1 -1
  86. package/dist/ops/cpu/adamAdjust.js +9 -9
  87. package/dist/ops/cpu/adamMoments.js +2 -2
  88. package/dist/ops/cpu/appendCache.js +2 -2
  89. package/dist/ops/cpu/attentionMask.js +5 -5
  90. package/dist/ops/cpu/fusedSoftmax.js +2 -2
  91. package/dist/ops/cpu/gatherSub.js +5 -5
  92. package/dist/ops/cpu/gelu.js +1 -1
  93. package/dist/ops/cpu/matMulGelu.js +2 -2
  94. package/dist/ops/cpu/matMulMul.js +1 -1
  95. package/dist/ops/cpu/mulDropout.js +1 -1
  96. package/dist/ops/cpu/normRMS.js +1 -1
  97. package/dist/ops/cpu/qkv.js +3 -3
  98. package/dist/ops/cpu/rope.js +5 -5
  99. package/dist/ops/cpu/scatterSub.js +7 -7
  100. package/dist/ops/fusedSoftmax.js +1 -1
  101. package/dist/ops/gatherSub.js +1 -1
  102. package/dist/ops/gelu.js +2 -2
  103. package/dist/ops/grads/attentionMask.js +1 -1
  104. package/dist/ops/grads/fusedSoftmax.js +2 -2
  105. package/dist/ops/grads/gelu.js +2 -2
  106. package/dist/ops/grads/matMulGelu.js +1 -1
  107. package/dist/ops/grads/normRMS.js +1 -1
  108. package/dist/ops/grads/qkv.js +1 -1
  109. package/dist/ops/grads/rope.js +1 -1
  110. package/dist/ops/matMulGelu.js +1 -1
  111. package/dist/ops/matMulMul.js +1 -1
  112. package/dist/ops/mulDrop.js +1 -1
  113. package/dist/ops/normRMS.js +1 -1
  114. package/dist/ops/qkv.js +1 -1
  115. package/dist/ops/rope.js +4 -4
  116. package/dist/ops/scatterSub.js +1 -1
  117. package/dist/ops/webgl/adamAdjust.js +2 -2
  118. package/dist/ops/webgl/adamMoments.js +1 -1
  119. package/dist/ops/webgl/appendCache.js +1 -1
  120. package/dist/ops/webgl/attentionMask.js +1 -1
  121. package/dist/ops/webgl/fusedSoftmax.js +4 -4
  122. package/dist/ops/webgl/gatherSub.js +1 -1
  123. package/dist/ops/webgl/gelu.js +2 -2
  124. package/dist/ops/webgl/log.js +3 -3
  125. package/dist/ops/webgl/matMulGelu.js +10 -10
  126. package/dist/ops/webgl/matMulMul.js +1 -1
  127. package/dist/ops/webgl/mulDropout.js +1 -1
  128. package/dist/ops/webgl/normRMS.js +2 -2
  129. package/dist/ops/webgl/qkv.js +1 -1
  130. package/dist/ops/webgl/rope.js +1 -1
  131. package/dist/ops/webgl/scatterSub.js +1 -1
  132. package/dist/ops/webgpu/adamAdjust.js +3 -3
  133. package/dist/ops/webgpu/adamMoments.js +3 -3
  134. package/dist/ops/webgpu/appendCache.js +3 -3
  135. package/dist/ops/webgpu/attentionMask.js +3 -3
  136. package/dist/ops/webgpu/gatherSub.js +3 -3
  137. package/dist/ops/webgpu/gelu.js +3 -3
  138. package/dist/ops/webgpu/normRMS.js +2 -2
  139. package/dist/ops/webgpu/normRMSGrad.js +5 -5
  140. package/dist/ops/webgpu/qkv.js +3 -3
  141. package/dist/ops/webgpu/rope.js +3 -3
  142. package/dist/ops/webgpu/scatterSub.js +3 -3
  143. package/dist/ops/webgpu/utils/reductions.js +4 -4
  144. package/dist/ops-542ai2vG.js +1525 -0
  145. package/dist/{random_width-sZORGo5k.js → random_width-DKGeiFuR.js} +1471 -1538
  146. package/dist/{range-CRuAh-gd.js → range-BcUvLuf5.js} +1 -1
  147. package/dist/{reciprocal-BvGAyKyu.js → reciprocal-DhDWSKiD.js} +1 -1
  148. package/dist/{register_all_kernels-BwDSRN-f.js → register_all_kernels-Do9VvZmo.js} +2488 -2534
  149. package/dist/{max-Ddnnb5xe.js → relu-B1AXs7p5.js} +6 -6
  150. package/dist/{reshape-CdBq1WJ6.js → reshape-WeJkT3ja.js} +1 -1
  151. package/dist/{scatter_nd_util-DUstGbU1.js → scatter_nd_util-B7yDhiQr.js} +1 -1
  152. package/dist/{selu_util-BJEXVvjX.js → selu_util-BgUO9gHY.js} +125 -146
  153. package/dist/{shared-wS99K7_n.js → shared-CZiWmQCI.js} +1 -1
  154. package/dist/{shared-B8ztnyEk.js → shared-V6D_md-c.js} +72 -72
  155. package/dist/{sin-BeA3tsEd.js → sin-CPxad7Am.js} +1 -1
  156. package/dist/{slice-BiOsknYS.js → slice-B7jXtPnp.js} +1 -1
  157. package/dist/{softmax-Bv_6lyMX.js → softmax-BfsyI4As.js} +1 -1
  158. package/dist/{split-B-dikLRw.js → split-BPxr8_8m.js} +1 -1
  159. package/dist/{stack-B17UN2nn.js → stack-BNwLzE43.js} +1 -1
  160. package/dist/{sum-66ew2byf.js → sum-ByFINZgi.js} +3 -3
  161. package/dist/{tensor-JwS7ZYY6.js → tensor-DbqgIV9B.js} +1 -1
  162. package/dist/tensor1d-CtJq5BOv.js +27 -0
  163. package/dist/{tensor2d-wxPAnDQy.js → tensor2d-CObBWBkW.js} +1 -1
  164. package/dist/tensor3d-BOukqWwr.js +30 -0
  165. package/dist/tensor4d-DLtk7Nxh.js +30 -0
  166. package/dist/training/Adam.js +2 -2
  167. package/dist/training/AdamExt.js +1 -1
  168. package/dist/training/DatasetBuilder.js +2 -2
  169. package/dist/training/Evaluator.d.ts +2 -2
  170. package/dist/training/FullTrainer.d.ts +3 -3
  171. package/dist/training/FullTrainer.js +61 -69
  172. package/dist/training/Trainer.d.ts +15 -3
  173. package/dist/training/Trainer.js +39 -47
  174. package/dist/training/sparseCrossEntropy.js +12 -13
  175. package/dist/utilities/arrayClose.d.ts +1 -1
  176. package/dist/utilities/arrayClose.js +16 -7
  177. package/dist/utilities/dummy.d.ts +4 -4
  178. package/dist/utilities/dummy.js +13 -13
  179. package/dist/utilities/multinomialCPU.js +2 -2
  180. package/dist/utilities/parameters.d.ts +1 -1
  181. package/dist/utilities/performance.js +1 -1
  182. package/dist/utilities/profile.js +1 -1
  183. package/dist/utilities/safetensors.js +2 -2
  184. package/dist/utilities/weights.js +2 -2
  185. package/dist/{variable-BuddVFLa.js → variable-DPFOJyRG.js} +1 -1
  186. package/dist/{webgpu_program-PFzf1hAQ.js → webgpu_program-Dhk9R5aG.js} +1 -1
  187. package/dist/{webgpu_util-D____QpY.js → webgpu_util-BqGnZg8t.js} +27 -27
  188. package/dist/{zeros--BdLQ3oG.js → zeros-Dnwix0p4.js} +1 -1
  189. package/package.json +2 -3
  190. package/dist/NanoGPTModel.d.ts +0 -52
  191. package/dist/NanoGPTModel.js +0 -203
  192. package/dist/TiedEmbedding-BxOerUmB.js +0 -43
  193. package/dist/ops-BFGCx8Ri.js +0 -1202
  194. package/dist/utilities/generate.d.ts +0 -3
  195. package/dist/utilities/generate.js +0 -22
  196. package/dist/utilities/save.d.ts +0 -9
  197. package/dist/utilities/save.js +0 -61
@@ -1,4 +1,4 @@
1
- import { f as l } from "../../index-BoWRt-10.js";
1
+ import { f as l } from "../../index-DdmHGZjq.js";
2
2
  class u {
3
3
  variableNames = ["labels", "logits", "values"];
4
4
  outputShape;
@@ -1,5 +1,5 @@
1
- import { f as a } from "../../index-BoWRt-10.js";
2
- import { u as s, C as x } from "../../kernel_funcs_utils-BYKWV8Aa.js";
1
+ import { f as a } from "../../index-DdmHGZjq.js";
2
+ import { u as s, C as x } from "../../kernel_funcs_utils-CDfFpUab.js";
3
3
  const t = 0.7978845608028654, r = 0.044715, c = x + `
4
4
  float x3 = x * x * x;
5
5
  float inner = x + ${r} * x3;
@@ -1,6 +1,6 @@
1
- import { f as e, af as r } from "../../index-BoWRt-10.js";
2
- import { u as s, l as N } from "../../kernel_funcs_utils-BYKWV8Aa.js";
3
- import { y as l } from "../../shared-wS99K7_n.js";
1
+ import { f as e, aj as r } from "../../index-DdmHGZjq.js";
2
+ import { u as s, l as N } from "../../kernel_funcs_utils-CDfFpUab.js";
3
+ import { y as l } from "../../shared-CZiWmQCI.js";
4
4
  /**
5
5
  * @license
6
6
  * Copyright 2020 Google LLC. All Rights Reserved.
@@ -1,7 +1,7 @@
1
- import { f as _, t as R, e as C, j as A, k as N, l as H, u as O } from "../../index-BoWRt-10.js";
2
- import { r as f } from "../../Reshape-DH5srBP0.js";
3
- import { M as U } from "../../mulmat_packed_gpu-VSekgsNv.js";
4
- import { m as E } from "../../mat_mul-8m8pfdcx.js";
1
+ import { f as _, t as R, e as C, j as A, l as N, n as H, u as O } from "../../index-DdmHGZjq.js";
2
+ import { r as f } from "../../Reshape-Bh_jzKzV.js";
3
+ import { M as U } from "../../mulmat_packed_gpu-q_Gmwyld.js";
4
+ import { m as E } from "../../mat_mul-Dpy2mMRu.js";
5
5
  const M = 0.7978845608028654, x = 0.044715, j = `
6
6
  vec4 x3 = x * x * x;
7
7
  vec4 inner = x + ${x} * x3;
@@ -34,7 +34,7 @@ function w({
34
34
  i === p,
35
35
  () => `Error in matMul: inner shapes (${i}) and (${p}) of Tensors with shapes ${e.shape} and ${t.shape} and transposeA=${s} and transposeB=${n} must match.`
36
36
  );
37
- const v = s ? [d, i, h] : [d, h, i], S = n ? [m, l, p] : [m, p, l], k = f({ inputs: { x: e }, backend: a, attrs: { shape: v } }), D = f({ inputs: { x: t }, backend: a, attrs: { shape: S } }), G = [k, D], y = Math.max(d, m), L = c, B = O(e.dtype, t.dtype), F = new U(
37
+ const v = s ? [d, i, h] : [d, h, i], S = n ? [m, l, p] : [m, p, l], D = f({ inputs: { x: e }, backend: a, attrs: { shape: v } }), G = f({ inputs: { x: t }, backend: a, attrs: { shape: S } }), g = [D, G], y = Math.max(d, m), L = c, B = O(e.dtype, t.dtype), F = new U(
38
38
  v,
39
39
  S,
40
40
  [y, h, l],
@@ -44,11 +44,11 @@ function w({
44
44
  L,
45
45
  !!o,
46
46
  !1
47
- ), g = [k, D];
48
- o && g.push(o);
49
- const $ = a.runWebGLProgram(F, g, B), I = f({ inputs: { x: $ }, backend: a, attrs: { shape: b } });
50
- G.push($);
51
- for (const P of G)
47
+ ), k = [D, G];
48
+ o && k.push(o);
49
+ const $ = a.runWebGLProgram(F, k, B), I = f({ inputs: { x: $ }, backend: a, attrs: { shape: b } });
50
+ g.push($);
51
+ for (const P of g)
52
52
  a.disposeIntermediateTensorInfo(P);
53
53
  return I;
54
54
  }
@@ -1,4 +1,4 @@
1
- import { f as u } from "../../index-BoWRt-10.js";
1
+ import { f as u } from "../../index-DdmHGZjq.js";
2
2
  import { batchMatMulGeluImpl as c } from "./matMulGelu.js";
3
3
  const M = `
4
4
  return a * b;
@@ -1,4 +1,4 @@
1
- import { f as m } from "../../index-BoWRt-10.js";
1
+ import { f as m } from "../../index-DdmHGZjq.js";
2
2
  class f {
3
3
  variableNames = ["a", "b"];
4
4
  outputShape;
@@ -1,5 +1,5 @@
1
- import { f as p, e as G } from "../../index-BoWRt-10.js";
2
- import { s as x } from "../../sum-66ew2byf.js";
1
+ import { f as p, e as G } from "../../index-DdmHGZjq.js";
2
+ import { s as x } from "../../sum-ByFINZgi.js";
3
3
  class y {
4
4
  variableNames = ["x", "meanSquare", "gamma"];
5
5
  outputShape;
@@ -1,4 +1,4 @@
1
- import { f as i } from "../../index-BoWRt-10.js";
1
+ import { f as i } from "../../index-DdmHGZjq.js";
2
2
  class l {
3
3
  variableNames = ["x", "kernel"];
4
4
  outputShape;
@@ -1,4 +1,4 @@
1
- import { f as u } from "../../index-BoWRt-10.js";
1
+ import { f as u } from "../../index-DdmHGZjq.js";
2
2
  class l {
3
3
  variableNames = ["x", "sin", "cos"];
4
4
  outputShape;
@@ -1,4 +1,4 @@
1
- import { f as i } from "../../index-BoWRt-10.js";
1
+ import { f as i } from "../../index-DdmHGZjq.js";
2
2
  class u {
3
3
  variableNames = ["labels", "softmaxProbs", "dy"];
4
4
  outputShape;
@@ -1,6 +1,6 @@
1
- import { g as p } from "../../webgpu_program-PFzf1hAQ.js";
2
- import { f as d, c as l } from "../../webgpu_util-D____QpY.js";
3
- import { f, a2 as c } from "../../index-BoWRt-10.js";
1
+ import { g as p } from "../../webgpu_program-Dhk9R5aG.js";
2
+ import { f as d, c as l } from "../../webgpu_util-BqGnZg8t.js";
3
+ import { f, a4 as c } from "../../index-DdmHGZjq.js";
4
4
  class h {
5
5
  variableNames = ["moments", "value"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { g as m } from "../../webgpu_program-PFzf1hAQ.js";
2
- import { f as u, c as p } from "../../webgpu_util-D____QpY.js";
3
- import { f as d, a2 as c } from "../../index-BoWRt-10.js";
1
+ import { g as m } from "../../webgpu_program-Dhk9R5aG.js";
2
+ import { f as u, c as p } from "../../webgpu_util-BqGnZg8t.js";
3
+ import { f as d, a4 as c } from "../../index-DdmHGZjq.js";
4
4
  class h {
5
5
  variableNames = ["moments", "gradient"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { g as u } from "../../webgpu_program-PFzf1hAQ.js";
2
- import { f as d, c as m } from "../../webgpu_util-D____QpY.js";
3
- import { f as l, a2 as f } from "../../index-BoWRt-10.js";
1
+ import { g as u } from "../../webgpu_program-Dhk9R5aG.js";
2
+ import { f as d, c as m } from "../../webgpu_util-BqGnZg8t.js";
3
+ import { f as l, a4 as f } from "../../index-DdmHGZjq.js";
4
4
  class g {
5
5
  variableNames = ["cache", "item"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { f, a2 as m } from "../../index-BoWRt-10.js";
2
- import { g as k } from "../../webgpu_program-PFzf1hAQ.js";
3
- import { f as l, c as v } from "../../webgpu_util-D____QpY.js";
1
+ import { f, a4 as m } from "../../index-DdmHGZjq.js";
2
+ import { g as k } from "../../webgpu_program-Dhk9R5aG.js";
3
+ import { f as l, c as v } from "../../webgpu_util-BqGnZg8t.js";
4
4
  class g {
5
5
  variableNames = ["q", "k"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { g as u } from "../../webgpu_program-PFzf1hAQ.js";
2
- import { f as h, c as p } from "../../webgpu_util-D____QpY.js";
3
- import { f as c, a2 as r } from "../../index-BoWRt-10.js";
1
+ import { g as u } from "../../webgpu_program-Dhk9R5aG.js";
2
+ import { f as h, c as p } from "../../webgpu_util-BqGnZg8t.js";
3
+ import { f as c, a4 as r } from "../../index-DdmHGZjq.js";
4
4
  class l {
5
5
  variableNames = ["labels", "logits", "values"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { f as i } from "../../index-BoWRt-10.js";
2
- import { g as o } from "../../webgpu_program-PFzf1hAQ.js";
3
- import { f as s, c as p } from "../../webgpu_util-D____QpY.js";
1
+ import { f as i } from "../../index-DdmHGZjq.js";
2
+ import { g as o } from "../../webgpu_program-Dhk9R5aG.js";
3
+ import { f as s, c as p } from "../../webgpu_util-BqGnZg8t.js";
4
4
  const u = 0.7978845608028654, a = 0.044715;
5
5
  class c {
6
6
  outputShape;
@@ -1,5 +1,5 @@
1
- import { f as n } from "../../webgpu_util-D____QpY.js";
2
- import { f as p, a2 as h } from "../../index-BoWRt-10.js";
1
+ import { f as n } from "../../webgpu_util-BqGnZg8t.js";
2
+ import { f as p, a4 as h } from "../../index-DdmHGZjq.js";
3
3
  import { createReduceInfo as u, reduce as c, createReductionShader as m } from "./utils/reductions.js";
4
4
  class d {
5
5
  outputShape;
@@ -1,9 +1,9 @@
1
- import { f, a2 as c, e as g } from "../../index-BoWRt-10.js";
1
+ import { f, a4 as c, e as g } from "../../index-DdmHGZjq.js";
2
2
  import { createReduceInfo as k } from "./utils/reductions.js";
3
- import { f as x } from "../../webgpu_util-D____QpY.js";
4
- import { g as z } from "../../webgpu_program-PFzf1hAQ.js";
5
- import { s as d } from "../../slice-BiOsknYS.js";
6
- import { s as w } from "../../sum-66ew2byf.js";
3
+ import { f as x } from "../../webgpu_util-BqGnZg8t.js";
4
+ import { g as z } from "../../webgpu_program-Dhk9R5aG.js";
5
+ import { s as d } from "../../slice-B7jXtPnp.js";
6
+ import { s as w } from "../../sum-ByFINZgi.js";
7
7
  class y {
8
8
  outputShape;
9
9
  shaderKey = "RMSNormGrad";
@@ -1,6 +1,6 @@
1
- import { g as h } from "../../webgpu_program-PFzf1hAQ.js";
2
- import { f as c, c as d } from "../../webgpu_util-D____QpY.js";
3
- import { f as p, a2 as m } from "../../index-BoWRt-10.js";
1
+ import { g as h } from "../../webgpu_program-Dhk9R5aG.js";
2
+ import { f as c, c as d } from "../../webgpu_util-BqGnZg8t.js";
3
+ import { f as p, a4 as m } from "../../index-DdmHGZjq.js";
4
4
  class l {
5
5
  variableNames = ["x", "kernel"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { g as c } from "../../webgpu_program-PFzf1hAQ.js";
2
- import { f as m, c as l } from "../../webgpu_util-D____QpY.js";
3
- import { f as x, a2 as f } from "../../index-BoWRt-10.js";
1
+ import { g as c } from "../../webgpu_program-Dhk9R5aG.js";
2
+ import { f as m, c as l } from "../../webgpu_util-BqGnZg8t.js";
3
+ import { f as x, a4 as f } from "../../index-DdmHGZjq.js";
4
4
  class S {
5
5
  variableNames = ["x", "sin", "cos"];
6
6
  outputShape;
@@ -1,6 +1,6 @@
1
- import { g as p } from "../../webgpu_program-PFzf1hAQ.js";
2
- import { f as u, c as d } from "../../webgpu_util-D____QpY.js";
3
- import { f as h, a2 as o } from "../../index-BoWRt-10.js";
1
+ import { g as p } from "../../webgpu_program-Dhk9R5aG.js";
2
+ import { f as u, c as d } from "../../webgpu_util-BqGnZg8t.js";
3
+ import { f as h, a4 as o } from "../../index-DdmHGZjq.js";
4
4
  class b {
5
5
  variableNames = ["labels", "softmaxProbs", "dy"];
6
6
  outputShape;
@@ -1,7 +1,7 @@
1
- import { p as l, j as d } from "../../../index-BoWRt-10.js";
2
- import { g as p } from "../../../webgpu_program-PFzf1hAQ.js";
3
- import { r as f } from "../../../Reshape-DvudQDvJ.js";
4
- import { c as x } from "../../../axis_util-BzbKo31C.js";
1
+ import { p as l, j as d } from "../../../index-DdmHGZjq.js";
2
+ import { g as p } from "../../../webgpu_program-Dhk9R5aG.js";
3
+ import { r as f } from "../../../Reshape-41YpQqEo.js";
4
+ import { c as x } from "../../../axis_util-Did9235A.js";
5
5
  function I(e, r, t, s, u) {
6
6
  return `
7
7
  fn DIV_CEIL(a : u32, b : u32) -> u32 {