@mni-ml/framework 0.0.1

package/dist/tensor_ops.js

@@ -0,0 +1,249 @@
+import { indexToPosition, toIndex, shapeProduct, broadcastIndex } from './tensor_data.js';
+import { Tensor } from './tensor.js';
+import { shapeBroadcast } from './tensor_data.js';
+export function tensorMap(fn) {
+    return (outStorage, outShape, outStrides, inStorage, inShape, inStrides) => {
+        const size = shapeProduct(outShape);
+        const outIndex = new Array(outShape.length).fill(0);
+        const inIndex = new Array(inShape.length).fill(0);
+        for (let ordinal = 0; ordinal < size; ordinal++) {
+            toIndex(ordinal, outShape, outIndex);
+            broadcastIndex(outIndex, outShape, inShape, inIndex);
+            const inPos = indexToPosition(inIndex, inStrides);
+            const outPos = indexToPosition(outIndex, outStrides);
+            outStorage[outPos] = fn(inStorage[inPos]);
+        }
+    };
+}
+export function tensorZip(fn) {
+    return (outStorage, outShape, outStrides, aStorage, aShape, aStrides, bStorage, bShape, bStrides) => {
+        const size = shapeProduct(outShape);
+        const outIndex = new Array(outShape.length).fill(0);
+        const aIndex = new Array(aShape.length).fill(0);
+        const bIndex = new Array(bShape.length).fill(0);
+        for (let ordinal = 0; ordinal < size; ordinal++) {
+            toIndex(ordinal, outShape, outIndex);
+            broadcastIndex(outIndex, outShape, aShape, aIndex);
+            broadcastIndex(outIndex, outShape, bShape, bIndex);
+            const aPos = indexToPosition(aIndex, aStrides);
+            const bPos = indexToPosition(bIndex, bStrides);
+            const outPos = indexToPosition(outIndex, outStrides);
+            outStorage[outPos] = fn(aStorage[aPos], bStorage[bPos]);
+        }
+    };
+}
+export function tensorReduce(fn) {
+    return (outStorage, outShape, outStrides, aStorage, aShape, aStrides, reduceDim) => {
+        const outSize = shapeProduct(outShape);
+        const reduceDimSize = aShape[reduceDim];
+        const outIndex = new Array(outShape.length).fill(0);
+        const aIndex = new Array(aShape.length).fill(0);
+        for (let ordinal = 0; ordinal < outSize; ordinal++) {
+            toIndex(ordinal, outShape, outIndex);
+            for (let i = 0; i < outShape.length; i++) {
+                aIndex[i] = outIndex[i];
+            }
+            const outPos = indexToPosition(outIndex, outStrides);
+            aIndex[reduceDim] = 0;
+            let acc = aStorage[indexToPosition(aIndex, aStrides)];
+            for (let j = 1; j < reduceDimSize; j++) {
+                aIndex[reduceDim] = j;
+                const aPos = indexToPosition(aIndex, aStrides);
+                acc = fn(acc, aStorage[aPos]);
+            }
+            outStorage[outPos] = acc;
+        }
+    };
+}
+/**
+ * Matrix multiply supporting 2D and 3D inputs with batch broadcasting.
+ * 2D inputs are padded to 3D internally; if both were 2D the output is squeezed back.
+ */
+export function tensorMatrixMultiply(A, B) {
+    // Index from end of tensor shape, such that length - 2 is rows, length -1 is cols
+    const [M, K] = [A.shape[A.shape.length - 2], A.shape[A.shape.length - 1]];
+    const [K2, N] = [B.shape[B.shape.length - 2], B.shape[B.shape.length - 1]];
+    if (!M || !K || !K2 || !N) {
+        return A;
+    }
+    if (K !== K2) {
+        throw new Error("A is of shape MxK. Expected B of shape K2xN");
+    }
+    let a = A;
+    let b = B;
+    // Make these always be exactly a 3 dimensional multiply, so the kernel only ever needs to deal with one batch loop + the 2D multiply
+    let Ais2D = false;
+    let Bis2D = false;
+    if (A.data.shape.length === 2) {
+        a = A.contiguous().view(1, M, K);
+        Ais2D = true;
+    }
+    if (B.data.shape.length === 2) {
+        b = B.contiguous().view(1, K2, N);
+        Bis2D = true;
+    }
+    // If both A and B had to be converted from 2D -> 3D, then we must remove a dimension at the end. Else it will simply just disappear as per mat mult
+    const both2D = Ais2D && Bis2D;
+    // Get resulting dimensions as array
+    const outShape = [...shapeBroadcast(a.shape.slice(0, -2), b.shape.slice(0, -2))];
+    outShape.push(M);
+    outShape.push(N);
+    let out = Tensor.zeros(outShape);
+    const size = shapeProduct(outShape);
+    const outIndex = new Array(outShape.length).fill(0);
+    const aIndex = new Array(a.shape.length).fill(0);
+    const bIndex = new Array(b.shape.length).fill(0);
+    for (let ordinal = 0; ordinal < size; ordinal++) {
+        toIndex(ordinal, outShape, outIndex);
+        broadcastIndex(outIndex, outShape, a.shape, aIndex);
+        broadcastIndex(outIndex, outShape, b.shape, bIndex);
+        let acc = 0;
+        for (let k = 0; k < K; k++) {
+            aIndex[aIndex.length - 1] = k; // K dim in A
+            bIndex[bIndex.length - 2] = k; // K dim in B
+            acc += a.get(aIndex) * b.get(bIndex);
+        }
+        out.set(outIndex, acc);
+    }
+    // Revert extra 3rd dimension
+    if (both2D) {
+        out = out.view(M, N);
+    }
+    return out;
+}
+/**
+ * Low-level 1D convolution kernel operating on raw Storage/Shape/Strides.
+ *
+ * Input shape:  [batch, in_channels, width]
+ * Weight shape: [out_channels, in_channels, kernel_width]
+ * Output shape: [batch, out_channels, out_width]  (caller pre-allocates)
+ *
+ * When reverse=false: output[b,oc,t] = sum_{ic,k} input[b,ic,t+k] * weight[oc,ic,k]
+ * When reverse=true:  output[b,oc,t] = sum_{ic,k} input[b,ic,t-k] * weight[oc,ic,k]
+ *
+ * Out-of-bounds input positions are treated as 0.
+ */
+export function _tensorConv1d(outStorage, outShape, outStrides, inputStorage, inputShape, inputStrides, weightStorage, weightShape, weightStrides, reverse) {
+    const outSize = shapeProduct(outShape);
+    const inChannels = inputShape[1];
+    const width = inputShape[2];
+    const kw = weightShape[2];
+    const outIndex = [0, 0, 0];
+    const inputIndex = [0, 0, 0];
+    const weightIndex = [0, 0, 0];
+    for (let ordinal = 0; ordinal < outSize; ordinal++) {
+        toIndex(ordinal, outShape, outIndex);
+        const b = outIndex[0];
+        const oc = outIndex[1];
+        const t = outIndex[2];
+        let val = 0;
+        for (let ic = 0; ic < inChannels; ic++) {
+            for (let k = 0; k < kw; k++) {
+                const s = reverse ? t - k : t + k;
+                if (s >= 0 && s < width) {
+                    inputIndex[0] = b;
+                    inputIndex[1] = ic;
+                    inputIndex[2] = s;
+                    weightIndex[0] = oc;
+                    weightIndex[1] = ic;
+                    weightIndex[2] = k;
+                    val += inputStorage[indexToPosition(inputIndex, inputStrides)]
+                        * weightStorage[indexToPosition(weightIndex, weightStrides)];
+                }
+            }
+        }
+        outStorage[indexToPosition(outIndex, outStrides)] = val;
+    }
+}
+/**
+ * 1D convolution: input [batch, in_channels, width] x weight [out_channels, in_channels, kw]
+ * -> output [batch, out_channels, width].
+ */
+export function tensorConv1d(input, weight, reverse = false) {
+    const batch = input.shape[0];
+    const inChannels = input.shape[1];
+    const width = input.shape[2];
+    const outChannels = weight.shape[0];
+    const weightInChannels = weight.shape[1];
+    if (inChannels !== weightInChannels) {
+        throw new Error(`Conv1d channel mismatch: input has ${inChannels} channels but weight expects ${weightInChannels}`);
+    }
+    const outShape = [batch, outChannels, width];
+    const out = Tensor.zeros(outShape);
+    _tensorConv1d(out.data.storage, out.data.shape, out.data.strides, input.data.storage, input.data.shape, input.data.strides, weight.data.storage, weight.data.shape, weight.data.strides, reverse);
+    return out;
+}
+/**
+ * Low-level 2D convolution kernel operating on raw Storage/Shape/Strides.
+ *
+ * Input shape:  [batch, in_channels, height, width]
+ * Weight shape: [out_channels, in_channels, kH, kW]
+ * Output shape: [batch, out_channels, out_height, out_width]  (caller pre-allocates)
+ *
+ * When reverse=false: output[b,oc,h,w] = sum_{ic,kh,kw} input[b,ic,h+kh,w+kw] * weight[oc,ic,kh,kw]
+ * When reverse=true:  output[b,oc,h,w] = sum_{ic,kh,kw} input[b,ic,h-kh,w-kw] * weight[oc,ic,kh,kw]
+ *
+ * Out-of-bounds input positions are treated as 0.
+ */
+export function _tensorConv2d(outStorage, outShape, outStrides, inputStorage, inputShape, inputStrides, weightStorage, weightShape, weightStrides, reverse) {
+    const outSize = shapeProduct(outShape);
+    const inChannels = inputShape[1];
+    const height = inputShape[2];
+    const width = inputShape[3];
+    const kH = weightShape[2];
+    const kW = weightShape[3];
+    const outIndex = [0, 0, 0, 0];
+    const inputIndex = [0, 0, 0, 0];
+    const weightIndex = [0, 0, 0, 0];
+    for (let ordinal = 0; ordinal < outSize; ordinal++) {
+        toIndex(ordinal, outShape, outIndex);
+        const b = outIndex[0];
+        const oc = outIndex[1];
+        const h = outIndex[2];
+        const w = outIndex[3];
+        let val = 0;
+        for (let ic = 0; ic < inChannels; ic++) {
+            for (let kh = 0; kh < kH; kh++) {
+                const sh = reverse ? h - kh : h + kh;
+                if (sh < 0 || sh >= height)
+                    continue;
+                for (let kw = 0; kw < kW; kw++) {
+                    const sw = reverse ? w - kw : w + kw;
+                    if (sw >= 0 && sw < width) {
+                        inputIndex[0] = b;
+                        inputIndex[1] = ic;
+                        inputIndex[2] = sh;
+                        inputIndex[3] = sw;
+                        weightIndex[0] = oc;
+                        weightIndex[1] = ic;
+                        weightIndex[2] = kh;
+                        weightIndex[3] = kw;
+                        val += inputStorage[indexToPosition(inputIndex, inputStrides)]
+                            * weightStorage[indexToPosition(weightIndex, weightStrides)];
+                    }
+                }
+            }
+        }
+        outStorage[indexToPosition(outIndex, outStrides)] = val;
+    }
+}
+/**
+ * 2D convolution: input [batch, in_channels, height, width] x weight [out_channels, in_channels, kH, kW]
+ * -> output [batch, out_channels, height, width].
+ */
+export function tensorConv2d(input, weight, reverse = false) {
+    const batch = input.shape[0];
+    const inChannels = input.shape[1];
+    const height = input.shape[2];
+    const width = input.shape[3];
+    const outChannels = weight.shape[0];
+    const weightInChannels = weight.shape[1];
+    if (inChannels !== weightInChannels) {
+        throw new Error(`Conv2d channel mismatch: input has ${inChannels} channels but weight expects ${weightInChannels}`);
+    }
+    const outShape = [batch, outChannels, height, width];
+    const out = Tensor.zeros(outShape);
+    _tensorConv2d(out.data.storage, out.data.shape, out.data.strides, input.data.storage, input.data.shape, input.data.strides, weight.data.storage, weight.data.shape, weight.data.strides, reverse);
+    return out;
+}
+//# sourceMappingURL=tensor_ops.js.map

package/dist/tensor_ops.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tensor_ops.js","sourceRoot":"","sources":["../src/tensor_ops.ts"],"names":[],"mappings":"AAMA,OAAO,EACH,eAAe,EACf,OAAO,EACP,YAAY,EACZ,cAAc,EACjB,MAAM,kBAAkB,CAAC;AAE1B,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAElD,MAAM,UAAU,SAAS,CACrB,EAAyB;IASzB,OAAO,CACH,UAAmB,EACnB,QAAe,EACf,UAAmB,EACnB,SAAkB,EAClB,OAAc,EACd,SAAkB,EACd,EAAE;QACN,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,OAAO,GAAa,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE5D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;YAC9C,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAErC,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAErD,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YAClD,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACrD,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,KAAK,CAAE,CAAC,CAAC;QAC/C,CAAC;IACL,CAAC,CAAC;AACN,CAAC;AAED,MAAM,UAAU,SAAS,CACrB,EAAoC;IAYpC,OAAO,CACH,UAAmB,EACnB,QAAe,EACf,UAAmB,EACnB,QAAiB,EACjB,MAAa,EACb,QAAiB,EACjB,QAAiB,EACjB,MAAa,EACb,QAAiB,EACb,EAAE;QACN,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACpC,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE1D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;YAC9C,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAErC,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YACnD,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;YAEnD,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC/C,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC/C,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACrD,UAAU,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAE,EAAE,QAAQ,CAAC,IAAI,CAAE,CAAC,CAAC;QAC9D,CAAC;IACL,CAAC,CAAA;AACL,CAAC;AAED,MAAM,UAAU,YAAY,CACxB,EAAsC;IAUtC,OAAO,CACH,UAAmB,EACnB,QAAe,EACf,UAAmB,EACnB,QAAiB,EACjB,MAAa,EACb,QAAiB,EACjB,SAAiB,EACb,EAAE;QACN,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACvC,MAAM,aAAa,GAAG,MAAM,CAAC,SAAS,CAAE,CAAC;QACzC,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;YACjD,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAErC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBACvC,MAAM,CAAC,CAAC,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;YAC7B,CAAC;YAED,MAAM,MAAM,GAAG,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YAErD,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;YACtB,IAAI,GAAG,GAAG,QAAQ,CAAC,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAE,CAAC;YAEvD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,EAAE,CAAC,EAAE,EAAE,CAAC;gBACrC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACtB,MAAM,IAAI,GAAG,eAAe,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;gBAC/C,GAAG,GAAG,EAAE,CAAC,GAAG,EAAE,QAAQ,CAAC,IAAI,CAAE,CAAC,CAAC;YACnC,CAAC;YAED,UAAU,CAAC,MAAM,CAAC,GAAG,GAAG,CAAC;QAC7B,CAAC;IACL,CAAC,CAAA;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAC,CAAS,EAAE,CAAS;IACrD,kFAAkF;IAClF,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;IAC1E,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;IAE3E,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC;QACxB,OAAO,CAAC,CAAC;IACb,CAAC;IAED,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;IACnE,CAAC;IAED,IAAI,CAAC,GAAW,CAAC,CAAC;IAClB,IAAI,CAAC,GAAW,CAAC,CAAC;IAElB,qIAAqI;IACrI,IAAI,KAAK,GAAG,KAAK,CAAC;IAClB,IAAI,KAAK,GAAG,KAAK,CAAC;IAClB,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACjC,KAAK,GAAG,IAAI,CAAC;IACjB,CAAC;IACD,IAAI,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;QAClC,KAAK,GAAG,IAAI,CAAC;IACjB,CAAC;IACD,oJAAoJ;IACpJ,MAAM,MAAM,GAAY,KAAK,IAAI,KAAK,CAAC;IAEvC,oCAAoC;IACpC,MAAM,QAAQ,GAAG,CAAC,GAAG,cAAc,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjF,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjB,IAAI,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAEjC,MAAM,IAAI,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAEpC,MAAM,QAAQ,GAAa,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC9D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAa,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAE3D,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC;QAC9C,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAErC,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpD,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAEpD,IAAI,GAAG,GAAG,CAAC,CAAC;QAEZ,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACzB,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,aAAa;YACjD,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,aAAa;YAEjD,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC;QAED,GAAG,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,6BAA6B;IAC7B,IAAI,MAAM,EAAE,CAAC;QACT,GAAG,GAAG,GAAG,CAAC,IAAI,CAAC,CAAC,EAAC,CAAC,CAAC,CAAC;IACxB,CAAC;IAED,OAAO,GAAG,CAAC;AACf,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,aAAa,CACzB,UAAmB,EAAE,QAAe,EAAE,UAAmB,EACzD,YAAqB,EAAE,UAAiB,EAAE,YAAqB,EAC/D,aAAsB,EAAE,WAAkB,EAAE,aAAsB,EAClE,OAAgB;IAEhB,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;IAClC,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;IAC7B,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAE,CAAC;IAE3B,MAAM,QAAQ,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC3B,MAAM,UAAU,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7B,MAAM,WAAW,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAE9B,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACjD,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QACvB,MAAM,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QACxB,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QAEvB,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC;YACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC1B,MAAM,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;gBAClC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,KAAK,EAAE,CAAC;oBACtB,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBAClB,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;oBACnB,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBAClB,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;oBACpB,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;oBACpB,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;oBACnB,GAAG,IAAI,YAAY,CAAC,eAAe,CAAC,UAAU,EAAE,YAAY,CAAC,CAAE;0BACxD,aAAa,CAAC,eAAe,CAAC,WAAW,EAAE,aAAa,CAAC,CAAE,CAAC;gBACvE,CAAC;YACL,CAAC;QACL,CAAC;QAED,UAAU,CAAC,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,GAAG,GAAG,CAAC;IAC5D,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CACxB,KAAa,EAAE,MAAc,EAAE,UAAmB,KAAK;IAEvD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IACnC,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IAC9B,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IACrC,MAAM,gBAAgB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IAE1C,IAAI,UAAU,KAAK,gBAAgB,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACX,sCAAsC,UAAU,gCAAgC,gBAAgB,EAAE,CACrG,CAAC;IACN,CAAC;IAED,MAAM,QAAQ,GAAU,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,CAAC;IACpD,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAEnC,aAAa,CACT,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,EAClD,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,EACxD,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,EAC3D,OAAO,CACV,CAAC;IAEF,OAAO,GAAG,CAAC;AACf,CAAC;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,aAAa,CACzB,UAAmB,EAAE,QAAe,EAAE,UAAmB,EACzD,YAAqB,EAAE,UAAiB,EAAE,YAAqB,EAC/D,aAAsB,EAAE,WAAkB,EAAE,aAAsB,EAClE,OAAgB;IAEhB,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;IAClC,MAAM,MAAM,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;IAC9B,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAE,CAAC;IAC7B,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAE,CAAC;IAC3B,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAE,CAAC;IAE3B,MAAM,QAAQ,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAC9B,MAAM,UAAU,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAChC,MAAM,WAAW,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;IAEjC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACjD,OAAO,CAAC,OAAO,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACrC,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QACvB,MAAM,EAAE,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QACxB,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QACvB,MAAM,CAAC,GAAG,QAAQ,CAAC,CAAC,CAAE,CAAC;QAEvB,IAAI,GAAG,GAAG,CAAC,CAAC;QACZ,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,UAAU,EAAE,EAAE,EAAE,EAAE,CAAC;YACrC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;gBAC7B,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;gBACrC,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,IAAI,MAAM;oBAAE,SAAS;gBACrC,KAAK,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;oBAC7B,MAAM,EAAE,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;oBACrC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,GAAG,KAAK,EAAE,CAAC;wBACxB,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;wBAClB,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;wBACnB,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;wBACnB,UAAU,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;wBACnB,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;wBACpB,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;wBACpB,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;wBACpB,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;wBACpB,GAAG,IAAI,YAAY,CAAC,eAAe,CAAC,UAAU,EAAE,YAAY,CAAC,CAAE;8BACxD,aAAa,CAAC,eAAe,CAAC,WAAW,EAAE,aAAa,CAAC,CAAE,CAAC;oBACvE,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QAED,UAAU,CAAC,eAAe,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,GAAG,GAAG,CAAC;IAC5D,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,YAAY,CACxB,KAAa,EAAE,MAAc,EAAE,UAAmB,KAAK;IAEvD,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IAC9B,MAAM,UAAU,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IACnC,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IAC/B,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IAC9B,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IACrC,MAAM,gBAAgB,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC;IAE1C,IAAI,UAAU,KAAK,gBAAgB,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CACX,sCAAsC,UAAU,gCAAgC,gBAAgB,EAAE,CACrG,CAAC;IACN,CAAC;IAED,MAAM,QAAQ,GAAU,CAAC,KAAK,EAAE,WAAW,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC;IAC5D,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAEnC,aAAa,CACT,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,EAClD,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,CAAC,OAAO,EACxD,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,OAAO,EAC3D,OAAO,CACV,CAAC;IAEF,OAAO,GAAG,CAAC;AACf,CAAC"}

package/package.json

@@ -0,0 +1,45 @@
+{
+  "name": "@mni-ml/framework",
+  "type": "module",
+  "version": "0.0.1",
+  "description": "A minimal machine learning library in TypeScript — autograd, tensors, neural network modules, and training utilities inspired by minitorch.",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/r-chong/TSTorch.git",
+    "directory": "packages/framework"
+  },
+  "keywords": [
+    "machine-learning",
+    "autograd",
+    "tensor",
+    "neural-network",
+    "deep-learning",
+    "typescript"
+  ],
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "devDependencies": {
+    "@fast-check/jest": "^2.1.1",
+    "@jest/globals": "^30.2.0",
+    "@types/jest": "^30.0.0",
+    "@types/node": "^25.0.8",
+    "jest": "^30.2.0",
+    "ts-jest": "^29.4.6",
+    "typescript": "^5.9.3"
+  },
+  "scripts": {
+    "test": "node --experimental-vm-modules node_modules/jest/bin/jest.js",
+    "build": "tsc"
+  },
+  "dependencies": {
+    "webgpu": "^0.3.8"
+  }
+}