@genai-fi/nanogpt 0.20.6 → 0.21.0

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 (208) hide show
  1. package/dist/{BaseTokeniser-DSg9zcYq.js → BaseTokeniser-C9TSv4th.js} +1 -1
  2. package/dist/{DatasetBuilder-DgURD85T.js → DatasetBuilder-B9UGNQT5.js} +82 -88
  3. package/dist/Generator.js +1 -1
  4. package/dist/{RealDiv-DBu0FQqT.js → RealDiv-CNsvC4AU.js} +6 -6
  5. package/dist/{Reshape-DqO3r8BC.js → Reshape-CwxdaU8n.js} +1 -1
  6. package/dist/{Reshape-CABOPB9d.js → Reshape-dnm9bO3B.js} +2 -2
  7. package/dist/TeachableLLM.js +1 -1
  8. package/dist/Trainer.js +1 -1
  9. package/dist/backend.js +2 -2
  10. package/dist/{backend_util-Cg-roD1p.js → backend_util-DAV0I34P.js} +5 -5
  11. package/dist/checks/appendCache.js +1 -1
  12. package/dist/checks/attentionMask.js +1 -1
  13. package/dist/checks/gelu.js +1 -1
  14. package/dist/checks/matMulGelu.js +1 -1
  15. package/dist/checks/normRMS.js +1 -1
  16. package/dist/checks/normRMSGrad.js +1 -1
  17. package/dist/checks/packUnpack.js +1 -1
  18. package/dist/checks/qkv.js +1 -1
  19. package/dist/checks/rope.js +6 -6
  20. package/dist/{chunk-BPntVaq0.js → chunk-CWhphoD1.js} +5 -5
  21. package/dist/{concat_util-CWDZCBlA.js → concat_util-CWzvQQlK.js} +1 -1
  22. package/dist/data/docx.js +58 -59
  23. package/dist/data/pdf.js +1 -1
  24. package/dist/data/textLoader.js +1 -1
  25. package/dist/{dist-DVmq73nz.js → dist-9wGF4ci9.js} +10 -10
  26. package/dist/{dist-VEU5mfO0.js → dist-BqAU9-yi.js} +4 -4
  27. package/dist/{dist-BewPQWjc.js → dist-Da20xy8E.js} +1874 -3071
  28. package/dist/{gelu-Bf1HW1RY.js → gelu-B6637MKa.js} +1 -1
  29. package/dist/{gpgpu_math-DvLcCH6u.js → gpgpu_math-DBYEAAdI.js} +2 -2
  30. package/dist/{kernel_funcs_utils-HiXOOx3f.js → kernel_funcs_utils-D-mATnGR.js} +3 -3
  31. package/dist/layers/BaseLayer.js +1 -1
  32. package/dist/layers/CausalSelfAttention.js +4 -4
  33. package/dist/layers/LoRA.js +3 -3
  34. package/dist/layers/MLP.js +2 -2
  35. package/dist/layers/PositionEmbedding.js +2 -2
  36. package/dist/layers/RMSNorm.js +1 -1
  37. package/dist/layers/RoPECache.js +1 -1
  38. package/dist/layers/TiedEmbedding.js +3 -3
  39. package/dist/layers/TransformerBlock.js +1 -1
  40. package/dist/layers/WeightStore.js +3 -3
  41. package/dist/loader/load.js +1 -1
  42. package/dist/loader/loadHF.js +1 -1
  43. package/dist/loader/loadTransformers.js +1 -1
  44. package/dist/loader/loadZipMeta.js +1 -1
  45. package/dist/loader/newZipLoad.js +1 -1
  46. package/dist/loader/oldZipLoad.js +1 -1
  47. package/dist/loader/save.d.ts +1 -0
  48. package/dist/loader/save.js +1 -1
  49. package/dist/{main-C0YUnw98.js → main-DdsoTJAW.js} +2012 -2015
  50. package/dist/main.js +1 -1
  51. package/dist/{matMul16-BNfZSnNM.js → matMul16-DGtlziMn.js} +3 -3
  52. package/dist/{matMulGelu-CPTntosE.js → matMulGelu-BAIgQaRx.js} +3 -3
  53. package/dist/models/NanoGPTV1.js +1 -1
  54. package/dist/models/NanoGPTV2.js +1 -1
  55. package/dist/models/factory.js +1 -1
  56. package/dist/models/model.js +1 -1
  57. package/dist/ops/adamAdjust.js +1 -1
  58. package/dist/ops/adamMoments.js +1 -1
  59. package/dist/ops/add16.js +1 -1
  60. package/dist/ops/appendCache.js +1 -1
  61. package/dist/ops/attentionMask.js +1 -1
  62. package/dist/ops/concat16.js +1 -1
  63. package/dist/ops/cpu/adamAdjust.js +1 -1
  64. package/dist/ops/cpu/adamMoments.js +1 -1
  65. package/dist/ops/cpu/appendCache.js +1 -1
  66. package/dist/ops/cpu/attentionMask.js +1 -1
  67. package/dist/ops/cpu/fusedSoftmax.js +1 -1
  68. package/dist/ops/cpu/gatherSub.js +1 -1
  69. package/dist/ops/cpu/gelu.js +1 -1
  70. package/dist/ops/cpu/matMul16.js +1 -1
  71. package/dist/ops/cpu/matMulGelu.js +2 -2
  72. package/dist/ops/cpu/matMulMul.js +1 -1
  73. package/dist/ops/cpu/mulDropout.js +1 -1
  74. package/dist/ops/cpu/normRMS.js +1 -1
  75. package/dist/ops/cpu/qkv.js +1 -1
  76. package/dist/ops/cpu/rope.js +1 -1
  77. package/dist/ops/cpu/scatterSub.js +1 -1
  78. package/dist/ops/dot16.js +2 -2
  79. package/dist/ops/dropout.js +1 -1
  80. package/dist/ops/dropout16.js +1 -1
  81. package/dist/ops/gatherSub.js +1 -1
  82. package/dist/ops/gelu.js +1 -1
  83. package/dist/ops/globalNorm.js +1 -1
  84. package/dist/ops/grads/add16.js +1 -1
  85. package/dist/ops/grads/attentionMask.js +2 -2
  86. package/dist/ops/grads/gelu.js +1 -1
  87. package/dist/ops/grads/matMul16.js +1 -1
  88. package/dist/ops/grads/matMulGelu.js +1 -1
  89. package/dist/ops/grads/normRMS.js +1 -1
  90. package/dist/ops/grads/pack16.js +1 -1
  91. package/dist/ops/grads/qkv.js +2 -2
  92. package/dist/ops/grads/rope.js +1 -1
  93. package/dist/ops/grads/softmax16.js +1 -1
  94. package/dist/ops/grads/unpack16.js +1 -1
  95. package/dist/ops/matMul16.js +1 -1
  96. package/dist/ops/matMulGelu.js +2 -2
  97. package/dist/ops/matMulMul.js +1 -1
  98. package/dist/ops/mul16.js +1 -1
  99. package/dist/ops/mulDrop.js +1 -1
  100. package/dist/ops/normRMS.js +1 -1
  101. package/dist/ops/pack16.js +1 -1
  102. package/dist/ops/qkv.js +1 -1
  103. package/dist/ops/reshape16.js +1 -1
  104. package/dist/ops/rope.js +1 -1
  105. package/dist/ops/scatterSub.js +1 -1
  106. package/dist/ops/slice16.js +1 -1
  107. package/dist/ops/softmax16.js +1 -1
  108. package/dist/ops/sub16.js +1 -1
  109. package/dist/ops/sum16.js +1 -1
  110. package/dist/ops/transpose16.js +1 -1
  111. package/dist/ops/unpack16.js +1 -1
  112. package/dist/ops/webgl/adamAdjust.js +2 -2
  113. package/dist/ops/webgl/adamMoments.js +3 -3
  114. package/dist/ops/webgl/appendCache.js +1 -1
  115. package/dist/ops/webgl/attentionMask.js +1 -1
  116. package/dist/ops/webgl/dropout16.js +1 -1
  117. package/dist/ops/webgl/fusedSoftmax.js +3 -3
  118. package/dist/ops/webgl/gatherSub.js +1 -1
  119. package/dist/ops/webgl/gelu.js +2 -2
  120. package/dist/ops/webgl/log.js +3 -3
  121. package/dist/ops/webgl/matMul16.js +8 -4
  122. package/dist/ops/webgl/matMulGelu.js +1 -1
  123. package/dist/ops/webgl/matMulMul.js +2 -2
  124. package/dist/ops/webgl/mulDropout.js +1 -1
  125. package/dist/ops/webgl/normRMS.js +1 -1
  126. package/dist/ops/webgl/qkv.js +1 -1
  127. package/dist/ops/webgl/rope.js +1 -1
  128. package/dist/ops/webgl/scatterSub.js +1 -1
  129. package/dist/ops/webgpu/adamAdjust.js +3 -3
  130. package/dist/ops/webgpu/adamMoments.js +3 -3
  131. package/dist/ops/webgpu/add16.js +1 -1
  132. package/dist/ops/webgpu/appendCache.js +3 -3
  133. package/dist/ops/webgpu/attentionMask.js +2 -2
  134. package/dist/ops/webgpu/attentionMask32_program.js +2 -2
  135. package/dist/ops/webgpu/clipScale.js +1 -1
  136. package/dist/ops/webgpu/concat16.js +5 -5
  137. package/dist/ops/webgpu/dropout16.js +3 -3
  138. package/dist/ops/webgpu/gatherSub.js +3 -3
  139. package/dist/ops/webgpu/gelu.js +3 -3
  140. package/dist/ops/webgpu/matMul16.js +8 -4
  141. package/dist/ops/webgpu/matMul16_program.js +7 -7
  142. package/dist/ops/webgpu/mul16.js +1 -1
  143. package/dist/ops/webgpu/norm2.js +1 -1
  144. package/dist/ops/webgpu/normRMS.js +3 -3
  145. package/dist/ops/webgpu/normRMSGrad.js +4 -4
  146. package/dist/ops/webgpu/pack16.js +1 -1
  147. package/dist/ops/webgpu/pack16_program.js +2 -2
  148. package/dist/ops/webgpu/qkv.js +2 -2
  149. package/dist/ops/webgpu/rope.js +3 -3
  150. package/dist/ops/webgpu/scatterSub.js +3 -3
  151. package/dist/ops/webgpu/slice16.js +4 -4
  152. package/dist/ops/webgpu/softmax16.js +1 -1
  153. package/dist/ops/webgpu/softmax16_program.js +2 -2
  154. package/dist/ops/webgpu/softmax16_subgroup_program.js +2 -2
  155. package/dist/ops/webgpu/softmax16grad.js +1 -1
  156. package/dist/ops/webgpu/sub16.js +1 -1
  157. package/dist/ops/webgpu/sum16.js +15 -15
  158. package/dist/ops/webgpu/transpose16.js +1 -1
  159. package/dist/ops/webgpu/transpose16_program.js +2 -2
  160. package/dist/ops/webgpu/transpose16_shared_program.js +3 -3
  161. package/dist/ops/webgpu/unpack16.js +3 -3
  162. package/dist/ops/webgpu/utils/binary_op.js +3 -3
  163. package/dist/ops/webgpu/utils/reductions.js +3 -3
  164. package/dist/{pack16-Ck-spx_F.js → pack16-BhuXNUS7.js} +1 -1
  165. package/dist/patches/webgpu_backend.js +3 -3
  166. package/dist/patches/webgpu_base.js +1 -1
  167. package/dist/patches/webgpu_program.js +4 -4
  168. package/dist/{pdf-UoDqCYzz.js → pdf-CdmW8ikl.js} +1159 -1160
  169. package/dist/{picomatch-3tUnMMbd.js → picomatch-BO4HbNfo.js} +125 -127
  170. package/dist/{rope-CbeGlsV8.js → rope-C8EAiSGY.js} +1 -1
  171. package/dist/{selu_util-zkAx5doH.js → selu_util-B_AFEfoz.js} +1 -1
  172. package/dist/{shared-D1coEFea.js → shared-24EoeHJl.js} +4 -4
  173. package/dist/{shared-DOgWaqvL.js → shared-CfTzpULd.js} +1 -1
  174. package/dist/{slice_util-Dgb3ANWI.js → slice_util-Dve4TkDn.js} +2 -2
  175. package/dist/{tfjs_backend-BjuQ5FqB.js → tfjs_backend-CydPRQTc.js} +2 -2
  176. package/dist/tokeniser/BaseTokeniser.js +1 -1
  177. package/dist/tokeniser/CharTokeniser.js +1 -1
  178. package/dist/tokeniser/bpe.js +1 -1
  179. package/dist/training/AdamW.js +1 -1
  180. package/dist/training/BasicTrainer.js +2 -2
  181. package/dist/training/DatasetBuilder.d.ts +0 -1
  182. package/dist/training/DatasetBuilder.js +2 -2
  183. package/dist/training/Evaluator.js +1 -1
  184. package/dist/training/PreTrainer.js +1 -1
  185. package/dist/training/SFTTrainer.js +1 -1
  186. package/dist/training/loss.js +2 -2
  187. package/dist/training/orthoGrad.js +1 -1
  188. package/dist/training/sparseCrossEntropy.js +1 -1
  189. package/dist/training/tasks/ConversationTask.js +1 -1
  190. package/dist/training/validation.js +1 -1
  191. package/dist/utilities/arrayClose.d.ts +1 -0
  192. package/dist/utilities/arrayClose.js +13 -1
  193. package/dist/utilities/dummy.js +1 -1
  194. package/dist/utilities/float16.d.ts +3 -0
  195. package/dist/utilities/float16.js +46 -0
  196. package/dist/utilities/multinomialCPU.js +1 -1
  197. package/dist/utilities/packed.js +1 -1
  198. package/dist/utilities/performance.js +1 -1
  199. package/dist/utilities/profile.js +1 -1
  200. package/dist/utilities/safetensors.d.ts +3 -1
  201. package/dist/utilities/safetensors.js +46 -36
  202. package/dist/utilities/sentences.js +1 -1
  203. package/dist/utilities/weights.js +1 -1
  204. package/dist/{webgpu-Dt7BMzWz.js → webgpu-B19Tw8Nd.js} +3 -3
  205. package/dist/{webgpu_program-WOyIVMlZ.js → webgpu_program-B0PokuRJ.js} +1 -1
  206. package/dist/{webgpu_util-B_F3SShA.js → webgpu_util-DfODg6XI.js} +1 -1
  207. package/package.json +5 -5
  208. package/dist/dist-DXwIvKxl.js +0 -896
package/dist/main.js CHANGED
@@ -1,4 +1,4 @@
1
- import { a as e, b as t, i as n, n as r, r as i, s as a, t as o, y as s } from "./main-C0YUnw98.js";
1
+ import { a as e, b as t, i as n, n as r, r as i, s as a, t as o, y as s } from "./main-DdsoTJAW.js";
2
2
  import c from "./tokeniser/CharTokeniser.js";
3
3
  import l from "./tokeniser/bpe.js";
4
4
  import { AdamWOptimizer as u } from "./training/AdamW.js";
@@ -1,12 +1,12 @@
1
- import { Fi as e, ii as t } from "./dist-BewPQWjc.js";
2
- import { t as n } from "./gelu-Bf1HW1RY.js";
1
+ import { Fi as e, ii as t } from "./dist-Da20xy8E.js";
2
+ import { t as n } from "./gelu-B6637MKa.js";
3
3
  import { isPackedTensor as r } from "./utilities/packed.js";
4
4
  import { transpose16 as i } from "./ops/transpose16.js";
5
5
  import { reshape16 as a } from "./ops/reshape16.js";
6
6
  import { mul16 as o } from "./ops/mul16.js";
7
7
  import "./ops/webgl/matMul16.js";
8
8
  import "./ops/cpu/matMul16.js";
9
- import { t as s } from "./pack16-Ck-spx_F.js";
9
+ import { t as s } from "./pack16-BhuXNUS7.js";
10
10
  //#region lib/ops/grads/matMul16.ts
11
11
  var c = {
12
12
  kernelName: "MatMul16",
@@ -1,6 +1,6 @@
1
- import { Ci as e, Ii as t, In as n, Ps as r, di as i, gr as a, ii as o, oc as s } from "./dist-BewPQWjc.js";
2
- import { a as c } from "./gpgpu_math-DvLcCH6u.js";
3
- import { t as l } from "./Reshape-CABOPB9d.js";
1
+ import { Ci as e, Ii as t, In as n, Ps as r, di as i, gr as a, ii as o, oc as s } from "./dist-Da20xy8E.js";
2
+ import { a as c } from "./gpgpu_math-DBYEAAdI.js";
3
+ import { t as l } from "./Reshape-dnm9bO3B.js";
4
4
  //#region node_modules/@tensorflow/tfjs-backend-webgl/dist/mulmat_packed_gpu.js
5
5
  var u = class {
6
6
  constructor(e, t, n, r = !1, i = !1, a = !1, o = null, s = !1, l = !1) {
@@ -1,2 +1,2 @@
1
- import { y as e } from "../main-C0YUnw98.js";
1
+ import { y as e } from "../main-DdsoTJAW.js";
2
2
  export { e as default };
@@ -1,2 +1,2 @@
1
- import { v as e } from "../main-C0YUnw98.js";
1
+ import { v as e } from "../main-DdsoTJAW.js";
2
2
  export { e as default };
@@ -1,2 +1,2 @@
1
- import { _ as e } from "../main-C0YUnw98.js";
1
+ import { _ as e } from "../main-DdsoTJAW.js";
2
2
  export { e as default };
@@ -1,2 +1,2 @@
1
- import { b as e } from "../main-C0YUnw98.js";
1
+ import { b as e } from "../main-DdsoTJAW.js";
2
2
  export { e as default };
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/adamAdjust.js";
3
3
  import "./webgl/adamAdjust.js";
4
4
  //#region lib/ops/adamAdjust.ts
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/adamMoments.js";
3
3
  import "./webgl/adamMoments.js";
4
4
  //#region lib/ops/adamMoments.ts
package/dist/ops/add16.js CHANGED
@@ -1,4 +1,4 @@
1
- import { ii as e, qr as t } from "../dist-BewPQWjc.js";
1
+ import { ii as e, qr as t } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../utilities/packed.js";
3
3
  import "./grads/add16.js";
4
4
  //#region lib/ops/add16.ts
@@ -1,4 +1,4 @@
1
- import { _r as e, ii as t, kt as n } from "../dist-BewPQWjc.js";
1
+ import { _r as e, ii as t, kt as n } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as r } from "../utilities/packed.js";
3
3
  import "./cpu/appendCache.js";
4
4
  import "./webgl/appendCache.js";
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/attentionMask.js";
3
3
  import "./webgl/attentionMask.js";
4
4
  import "./grads/attentionMask.js";
@@ -1,4 +1,4 @@
1
- import { _r as e, ii as t } from "../dist-BewPQWjc.js";
1
+ import { _r as e, ii as t } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../utilities/packed.js";
3
3
  //#region lib/ops/concat16.ts
4
4
  function r(r, i) {
@@ -1,4 +1,4 @@
1
- import { Gr as e, Ii as t, Wr as n, gn as r, qr as i } from "../../dist-BewPQWjc.js";
1
+ import { Gr as e, Ii as t, Wr as n, gn as r, qr as i } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/adamAdjust.ts
3
3
  function a(t) {
4
4
  let { moments: a, value: o } = t.inputs, { beta1: s, beta2: c, epsilon: l, learningRate: u } = t.attrs, d = a.shape.length, f = Array(d).fill(0), p = a.shape.slice();
@@ -1,4 +1,4 @@
1
- import { Ii as e, V as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, V as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/adamMoments.ts
3
3
  function n(e) {
4
4
  let { moments: n, gradient: r } = e.inputs, { beta1: i, beta2: a } = e.attrs, o = n.shape.length, s = Array(o).fill(0), c = n.shape.slice();
@@ -1,4 +1,4 @@
1
- import { Ii as e, _r as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, _r as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/appendCache.ts
3
3
  function n(e) {
4
4
  let { cache: n, item: r } = e.inputs, { maxSize: i, pastLen: a } = e.attrs, o = n.shape[0], s = n.shape[1], c = n.shape[3], l = r.shape[2];
@@ -1,4 +1,4 @@
1
- import { Ii as e, Ot as t, Pn as n, _n as r, ar as i, gr as a, kt as o, p as s } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, Ot as t, Pn as n, _n as r, ar as i, gr as a, kt as o, p as s } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/attentionMask.ts
3
3
  function c(e) {
4
4
  let { q: c, k: l } = e.inputs, { divisor: u } = e.attrs, d = c.shape[2], f = l.shape[2], p = n(s.bandPart(t([f, f]), -1, 0).cast("bool"), o([f, f]), i([f, f], -Infinity)), m = a(c, l, !1, !0).mul(r(u)), h = p.slice([0, 0], [d, f]).expandDims(0).expandDims(0);
@@ -1,4 +1,4 @@
1
- import { Ii as e, Y as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, Y as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/fusedSoftmax.ts
3
3
  function n(e) {
4
4
  let { inputs: n, attrs: r } = e, { logits: i } = n, { dim: a, dropoutRate: o } = r;
@@ -1,4 +1,4 @@
1
- import { Bt as e, Ii as t, S as n, V as r, dt as i } from "../../dist-BewPQWjc.js";
1
+ import { Bt as e, Ii as t, S as n, V as r, dt as i } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/gatherSub.ts
3
3
  function a(t) {
4
4
  let { values: a, labels: o, logits: s } = t.inputs, c = o.shape[0];
@@ -1,4 +1,4 @@
1
- import { Ii as e, di as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, di as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/gelu.ts
3
3
  var n = .7978845608028654, r = .044715;
4
4
  function i(e) {
@@ -1,4 +1,4 @@
1
- import { Ii as e, gr as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, gr as t } from "../../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../../utilities/packed.js";
3
3
  //#region lib/ops/cpu/matMul16.ts
4
4
  function r(e) {
@@ -1,5 +1,5 @@
1
- import { Ii as e, di as t, gr as n } from "../../dist-BewPQWjc.js";
2
- import { n as r, t as i } from "../../gelu-Bf1HW1RY.js";
1
+ import { Ii as e, di as t, gr as n } from "../../dist-Da20xy8E.js";
2
+ import { n as r, t as i } from "../../gelu-B6637MKa.js";
3
3
  //#region lib/ops/cpu/matMulGelu.ts
4
4
  function a(e) {
5
5
  let { inputs: i } = e, { x: a, kernel: o } = i, s = a, c = o;
@@ -1,4 +1,4 @@
1
- import { Ii as e, di as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, di as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/matMulMul.ts
3
3
  function n(e) {
4
4
  let { inputs: n, attrs: r } = e, { transposeA: i, transposeB: a } = r, { x: o, kernel: s, y: c } = n, l = o, u = s, d = c;
@@ -1,4 +1,4 @@
1
- import { Ii as e, Wr as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, Wr as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/mulDropout.ts
3
3
  function n(e) {
4
4
  let { inputs: n } = e, { a: r, b: i } = n;
@@ -1,4 +1,4 @@
1
- import { Ii as e, di as t } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, di as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/normRMS.ts
3
3
  function n(e) {
4
4
  let { inputs: n } = e, { x: r, gamma: i } = n, a = r, o = i;
@@ -1,4 +1,4 @@
1
- import { G as e, Ii as t, br as n } from "../../dist-BewPQWjc.js";
1
+ import { G as e, Ii as t, br as n } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/qkv.ts
3
3
  function r(t) {
4
4
  let { x: r, kernel: i } = t.inputs, { heads: a, packed: o } = t.attrs;
@@ -1,4 +1,4 @@
1
- import { Ii as e, V as t, _r as n, an as r, dt as i } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, V as t, _r as n, an as r, dt as i } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/rope.ts
3
3
  function a(e, a, o, s, c) {
4
4
  let l = s.shape[3], u = o;
@@ -1,4 +1,4 @@
1
- import { Bt as e, C as t, Ii as n, Ot as r, V as i, Wr as a, dt as o } from "../../dist-BewPQWjc.js";
1
+ import { Bt as e, C as t, Ii as n, Ot as r, V as i, Wr as a, dt as o } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/cpu/scatterSub.ts
3
3
  function s(n) {
4
4
  let { logits: s, labels: c, dy: l } = n.inputs, u = c.shape[0], d = s.shape[1];
package/dist/ops/dot16.js CHANGED
@@ -1,8 +1,8 @@
1
- import { o as e } from "../tfjs_backend-BjuQ5FqB.js";
1
+ import { o as e } from "../tfjs_backend-CydPRQTc.js";
2
2
  import { isPackedTensor as t } from "../utilities/packed.js";
3
3
  import { transpose16 as n } from "./transpose16.js";
4
4
  import { reshape16 as r } from "./reshape16.js";
5
- import { t as i } from "../matMul16-BNfZSnNM.js";
5
+ import { t as i } from "../matMul16-DGtlziMn.js";
6
6
  //#region lib/ops/dot16.ts
7
7
  function a(a, o, s = !1, c = !1) {
8
8
  if (!t(a) && !t(o)) return e(a, o);
@@ -1,4 +1,4 @@
1
- import { Gr as e, Ps as t, ft as n, mi as r, on as i, qr as a, wi as o, x as s } from "../dist-BewPQWjc.js";
1
+ import { Gr as e, Ps as t, ft as n, mi as r, on as i, qr as a, wi as o, x as s } from "../dist-Da20xy8E.js";
2
2
  import { mul16 as c } from "./mul16.js";
3
3
  //#region lib/ops/dropout.ts
4
4
  function l(l, u, d, f) {
@@ -1,4 +1,4 @@
1
- import { Fi as e, ii as t } from "../dist-BewPQWjc.js";
1
+ import { Fi as e, ii as t } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../utilities/packed.js";
3
3
  import "./webgl/dropout16.js";
4
4
  //#region lib/ops/grads/dropout16.ts
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/gatherSub.js";
3
3
  import "./webgl/gatherSub.js";
4
4
  //#region lib/ops/gatherSub.ts
package/dist/ops/gelu.js CHANGED
@@ -1,2 +1,2 @@
1
- import { n as e, t } from "../gelu-Bf1HW1RY.js";
1
+ import { n as e, t } from "../gelu-B6637MKa.js";
2
2
  export { t as dGelu, e as gelu };
@@ -1,4 +1,4 @@
1
- import { di as e, ii as t, kt as n } from "../dist-BewPQWjc.js";
1
+ import { di as e, ii as t, kt as n } from "../dist-Da20xy8E.js";
2
2
  //#region lib/ops/globalNorm.ts
3
3
  function r(r, i, a = 1) {
4
4
  return e(() => {
@@ -1,4 +1,4 @@
1
- import { Fi as e, In as t, Rn as n } from "../../dist-BewPQWjc.js";
1
+ import { Fi as e, In as t, Rn as n } from "../../dist-Da20xy8E.js";
2
2
  import { reshape16 as r } from "../reshape16.js";
3
3
  import { sum16 as i } from "../sum16.js";
4
4
  //#region lib/ops/grads/add16.ts
@@ -1,6 +1,6 @@
1
- import { Fi as e } from "../../dist-BewPQWjc.js";
1
+ import { Fi as e } from "../../dist-Da20xy8E.js";
2
2
  import { transpose16 as t } from "../transpose16.js";
3
- import { a as n } from "../../matMul16-BNfZSnNM.js";
3
+ import { a as n } from "../../matMul16-DGtlziMn.js";
4
4
  //#region lib/ops/grads/attentionMask.ts
5
5
  e({
6
6
  kernelName: "AttentionMask",
@@ -1,2 +1,2 @@
1
- import { r as e } from "../../gelu-Bf1HW1RY.js";
1
+ import { r as e } from "../../gelu-B6637MKa.js";
2
2
  export { e as geluGradConfig };
@@ -1,2 +1,2 @@
1
- import { o as e } from "../../matMul16-BNfZSnNM.js";
1
+ import { o as e } from "../../matMul16-DGtlziMn.js";
2
2
  export { e as matMul16GradConfig };
@@ -1,4 +1,4 @@
1
- import { Fi as e, ii as t } from "../../dist-BewPQWjc.js";
1
+ import { Fi as e, ii as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/grads/matMulGelu.ts
3
3
  function n(e, n, r) {
4
4
  return t().runKernel("MatMulGeluGrad", {
@@ -1,4 +1,4 @@
1
- import { Fi as e, ii as t } from "../../dist-BewPQWjc.js";
1
+ import { Fi as e, ii as t } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/grads/normRMS.ts
3
3
  function n(e, n, r) {
4
4
  return t().runKernel("RMSNormGrad", r ? {
@@ -1,2 +1,2 @@
1
- import { n as e } from "../../pack16-Ck-spx_F.js";
1
+ import { n as e } from "../../pack16-BhuXNUS7.js";
2
2
  export { e as packGradConfig };
@@ -1,6 +1,6 @@
1
- import { Fi as e, H as t } from "../../dist-BewPQWjc.js";
1
+ import { Fi as e, H as t } from "../../dist-Da20xy8E.js";
2
2
  import { sum16 as n } from "../sum16.js";
3
- import { o as r } from "../../matMul16-BNfZSnNM.js";
3
+ import { o as r } from "../../matMul16-DGtlziMn.js";
4
4
  import { concat16 as i } from "../concat16.js";
5
5
  //#region lib/ops/grads/qkv.ts
6
6
  var a = {
@@ -1,2 +1,2 @@
1
- import { n as e } from "../../rope-CbeGlsV8.js";
1
+ import { n as e } from "../../rope-C8EAiSGY.js";
2
2
  export { e as ropeGradConfig };
@@ -1,4 +1,4 @@
1
- import { Fi as e, ii as t } from "../../dist-BewPQWjc.js";
1
+ import { Fi as e, ii as t } from "../../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../../utilities/packed.js";
3
3
  //#region lib/ops/grads/softmax16.ts
4
4
  function r(e, n) {
@@ -1,2 +1,2 @@
1
- import { i as e } from "../../pack16-Ck-spx_F.js";
1
+ import { i as e } from "../../pack16-BhuXNUS7.js";
2
2
  export { e as unpackGradConfig };
@@ -1,2 +1,2 @@
1
- import { a as e, i as t, n, r, t as i } from "../matMul16-BNfZSnNM.js";
1
+ import { a as e, i as t, n, r, t as i } from "../matMul16-DGtlziMn.js";
2
2
  export { i as matMul16, n as matMul16Gelu, r as matMul16ScaleA, t as matMul16ScaleB, e as matMul16Scaled };
@@ -1,5 +1,5 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
2
- import "../matMulGelu-CPTntosE.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
+ import "../matMulGelu-BAIgQaRx.js";
3
3
  import "./cpu/matMulGelu.js";
4
4
  import "./grads/matMulGelu.js";
5
5
  //#region lib/ops/matMulGelu.ts
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/matMulMul.js";
3
3
  import "./webgl/matMulMul.js";
4
4
  //#region lib/ops/matMulMul.ts
package/dist/ops/mul16.js CHANGED
@@ -1,4 +1,4 @@
1
- import { Fi as e, In as t, Rn as n, Wr as r, ii as i } from "../dist-BewPQWjc.js";
1
+ import { Fi as e, In as t, Rn as n, Wr as r, ii as i } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as a } from "../utilities/packed.js";
3
3
  import { reshape16 as o } from "./reshape16.js";
4
4
  import { sum16 as s } from "./sum16.js";
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/mulDropout.js";
3
3
  import "./webgl/mulDropout.js";
4
4
  //#region lib/ops/mulDrop.ts
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/normRMS.js";
3
3
  import "./webgl/normRMS.js";
4
4
  import "./grads/normRMS.js";
@@ -1,2 +1,2 @@
1
- import { t as e } from "../pack16-Ck-spx_F.js";
1
+ import { t as e } from "../pack16-BhuXNUS7.js";
2
2
  export { e as pack16 };
package/dist/ops/qkv.js CHANGED
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/qkv.js";
3
3
  import "./webgl/qkv.js";
4
4
  import "./grads/qkv.js";
@@ -1,4 +1,4 @@
1
- import { Fi as e, Ii as t, br as n, ii as r } from "../dist-BewPQWjc.js";
1
+ import { Fi as e, Ii as t, br as n, ii as r } from "../dist-Da20xy8E.js";
2
2
  //#region lib/ops/reshape16.ts
3
3
  e({
4
4
  kernelName: "Reshape16",
package/dist/ops/rope.js CHANGED
@@ -1,2 +1,2 @@
1
- import { t as e } from "../rope-CbeGlsV8.js";
1
+ import { t as e } from "../rope-C8EAiSGY.js";
2
2
  export { e as rope };
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import "./cpu/scatterSub.js";
3
3
  import "./webgl/scatterSub.js";
4
4
  //#region lib/ops/scatterSub.ts
@@ -1,4 +1,4 @@
1
- import { ii as e, mr as t } from "../dist-BewPQWjc.js";
1
+ import { ii as e, mr as t } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../utilities/packed.js";
3
3
  //#region lib/ops/slice16.ts
4
4
  function r(r, i, a) {
@@ -1,4 +1,4 @@
1
- import { ii as e } from "../dist-BewPQWjc.js";
1
+ import { ii as e } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as t } from "../utilities/packed.js";
3
3
  import "./grads/softmax16.js";
4
4
  //#region lib/ops/softmax16.ts
package/dist/ops/sub16.js CHANGED
@@ -1,4 +1,4 @@
1
- import { Bt as e, ii as t } from "../dist-BewPQWjc.js";
1
+ import { Bt as e, ii as t } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../utilities/packed.js";
3
3
  //#region lib/ops/sub16.ts
4
4
  function r(r, i) {
package/dist/ops/sum16.js CHANGED
@@ -1,4 +1,4 @@
1
- import { ii as e, mn as t } from "../dist-BewPQWjc.js";
1
+ import { ii as e, mn as t } from "../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as n } from "../utilities/packed.js";
3
3
  //#region lib/ops/sum16.ts
4
4
  function r(r, i, a = !1) {
@@ -1,4 +1,4 @@
1
- import { Fi as e, Ii as t, On as n, ii as r, w as i } from "../dist-BewPQWjc.js";
1
+ import { Fi as e, Ii as t, On as n, ii as r, w as i } from "../dist-Da20xy8E.js";
2
2
  import { forceFloat as a, forcePacked as o } from "./grads/utils.js";
3
3
  import { isPackedTensor as s } from "../utilities/packed.js";
4
4
  //#region lib/ops/transpose16.ts
@@ -1,2 +1,2 @@
1
- import { r as e } from "../pack16-Ck-spx_F.js";
1
+ import { r as e } from "../pack16-BhuXNUS7.js";
2
2
  export { e as unpack16 };
@@ -1,5 +1,5 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
2
- import { t } from "../../Reshape-CABOPB9d.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
+ import { t } from "../../Reshape-dnm9bO3B.js";
3
3
  //#region lib/ops/webgl/adamAdjust.ts
4
4
  var n = class {
5
5
  variableNames = ["moments", "value"];
@@ -1,4 +1,4 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/webgl/adamMoments.ts
3
3
  var t = class {
4
4
  variableNames = ["moments", "gradient"];
@@ -19,8 +19,8 @@ var t = class {
19
19
  float m = getMomentsAtOutCoords();
20
20
  ${n} coords = getOutputCoords();
21
21
 
22
- // Add gradient clipping here
23
- float g = clamp(getGradient(${i}), -1.0, 1.0);
22
+ // TODO: Add scaling
23
+ float g = getGradient(${i});
24
24
  int which = ${r};
25
25
 
26
26
  float beta = which == 0 ? beta1 : beta2;
@@ -1,4 +1,4 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/webgl/appendCache.ts
3
3
  var t = class {
4
4
  variableNames = ["cache", "item"];
@@ -1,4 +1,4 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/webgl/attentionMask.ts
3
3
  var t = class {
4
4
  variableNames = ["q", "k"];
@@ -1,4 +1,4 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/webgl/dropout16.ts
3
3
  function t(e) {
4
4
  let { x: t } = e.inputs;
@@ -1,6 +1,6 @@
1
- import { Ii as e, Tn as t, nc as n } from "../../dist-BewPQWjc.js";
2
- import { t as r } from "../../Reshape-CABOPB9d.js";
3
- import { a as i, r as a, t as o } from "../../RealDiv-DBu0FQqT.js";
1
+ import { Ii as e, Tn as t, nc as n } from "../../dist-Da20xy8E.js";
2
+ import { t as r } from "../../Reshape-dnm9bO3B.js";
3
+ import { a as i, r as a, t as o } from "../../RealDiv-CNsvC4AU.js";
4
4
  //#region lib/ops/webgl/fusedSoftmax.ts
5
5
  var s = class {
6
6
  variableNames = ["logits", "maxLogits"];
@@ -1,4 +1,4 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/webgl/gatherSub.ts
3
3
  var t = class {
4
4
  variableNames = [
@@ -1,5 +1,5 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
2
- import { _ as t, i as n } from "../../kernel_funcs_utils-HiXOOx3f.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
+ import { _ as t, i as n } from "../../kernel_funcs_utils-D-mATnGR.js";
3
3
  //#region lib/ops/webgl/gelu.ts
4
4
  var r = .7978845608028654, i = .044715, a = n({ opSnippet: t + `
5
5
  float x3 = x * x * x;
@@ -1,6 +1,6 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
2
- import { y as t } from "../../shared-DOgWaqvL.js";
3
- import { i as n, t as r } from "../../kernel_funcs_utils-HiXOOx3f.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
+ import { y as t } from "../../shared-CfTzpULd.js";
3
+ import { i as n, t as r } from "../../kernel_funcs_utils-D-mATnGR.js";
4
4
  //#region lib/ops/webgl/log.ts
5
5
  e({
6
6
  kernelName: "Log",
@@ -1,4 +1,4 @@
1
- import { Ii as e, Wr as t, _n as n, br as r, gr as i, hn as a, lt as o, w as s } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, Wr as t, _n as n, br as r, gr as i, hn as a, lt as o, w as s } from "../../dist-Da20xy8E.js";
2
2
  import { isPackedTensor as c } from "../../utilities/packed.js";
3
3
  import { matMulMul as l } from "../matMulMul.js";
4
4
  import { matMulGelu as u } from "../matMulGelu.js";
@@ -7,12 +7,16 @@ function d(e) {
7
7
  let { A: d, B: f } = e.inputs, { transposeA: p, transposeB: m, scale: h, activation: g, scaleA: _, scaleB: v, forceOutputShape: y, perm: b } = e.attrs, x = !c(d), S = !c(f);
8
8
  if (x && S) {
9
9
  let e = _ === void 0 ? d : t(d, n(_)), c = v === void 0 ? f : t(f, n(v)), x;
10
- if (h !== void 0) x = l(e, c, n(h), p, m);
11
- else if (g === "gelu") x = u(e, c);
10
+ if (g === "gelu") x = u(e, c);
12
11
  else if (g === "relu2") {
13
12
  let t = i(e, c, p, m), n = o(t);
14
13
  t.dispose(), x = a(n), n.dispose();
15
- } else x = g === "relu" ? o(i(e, c, p, m)) : i(e, c, p, m);
14
+ } else if (g === "relu") {
15
+ if (x = o(i(e, c, p, m)), h !== void 0) {
16
+ let e = t(x, n(h));
17
+ x.dispose(), x = e;
18
+ }
19
+ } else x = h === void 0 ? i(e, c, p, m) : l(e, c, n(h), p, m);
16
20
  if (b) if (y) {
17
21
  let e = r(x, y);
18
22
  x.dispose();
@@ -1,2 +1,2 @@
1
- import { n as e, r as t, t as n } from "../../matMulGelu-CPTntosE.js";
1
+ import { n as e, r as t, t as n } from "../../matMulGelu-BAIgQaRx.js";
2
2
  export { n as MATMUL_SHARED_DIM_THRESHOLD, e as batchMatMulGeluImpl, t as batchMatMulKernel };
@@ -1,5 +1,5 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
2
- import { n as t } from "../../matMulGelu-CPTntosE.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
+ import { n as t } from "../../matMulGelu-BAIgQaRx.js";
3
3
  //#region lib/ops/webgl/matMulMul.ts
4
4
  var n = "\n return a * b;\n";
5
5
  function r(e) {
@@ -1,4 +1,4 @@
1
- import { Ii as e } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/webgl/mulDropout.ts
3
3
  var t = class {
4
4
  variableNames = ["a", "b"];
@@ -1,4 +1,4 @@
1
- import { Ii as e, ii as t, mn as n } from "../../dist-BewPQWjc.js";
1
+ import { Ii as e, ii as t, mn as n } from "../../dist-Da20xy8E.js";
2
2
  //#region lib/ops/webgl/normRMS.ts
3
3
  var r = class {
4
4
  variableNames = ["x", "meanSquare"];