@isidorus/cpu 0.0.0-alpha.1 → 0.0.0-alpha.2

package/dist/graph.d.ts

@@ -21,6 +21,9 @@ export type AttrValue = {
 } | {
     kind: "list_int";
     value: number[];
+} | {
+    kind: "string";
+    value: string;
 } | {
     kind: "tensor";
     value: InlineTensor;
@@ -58,9 +61,30 @@ export declare class Graph {
      * @param inputs    Output references from prior ops
      * @param attrs     Op attributes
      * @param name      Optional explicit op name (auto-generated if omitted)
+     * @param controlInputs Op names that must complete before this op runs.
+     *                      Used by globalVariablesInitializer to sequence init
+     *                      ops before the NoOp target that callers wait on.
      * @returns         Array of output Tensors (one per op output)
      */
-    addOp(type: string, inputs: TFOutput[], attrs?: Record<string, AttrValue>, name?: string): Tensor[];
+    addOp(type: string, inputs: TFOutput[], attrs?: Record<string, AttrValue>, name?: string, controlInputs?: string[]): Tensor[];
+    /**
+     * addGradients — compute symbolic gradients via TF_AddGradients.
+     *
+     * Injects gradient ops into the graph.  Returns one gradient Tensor per
+     * entry in `x` — the partial derivative dSum(y)/dx_i.
+     *
+     * @param y   Loss outputs to differentiate (typically a scalar loss tensor)
+     * @param x   Parameters to differentiate with respect to
+     * @param dx  Initial upstream gradients (default: ones, i.e. dL/dy = 1)
+     *
+     * @example
+     * const loss = ops.sparseSoftmaxCrossEntropyWithLogits(g, labels, logits);
+     * const wVal = ops.readVariable(g, wHandle, DType.FLOAT32);
+     * const [dw] = g.addGradients([loss], [wVal]);
+     * // dw is now a Tensor representing dLoss/dW
+     * // pass it to applyGradientDescent or applyAdam
+     */
+    addGradients(y: TFOutput[], x: TFOutput[], dx?: TFOutput[]): Tensor[];
     /** Whether an op with the given name exists in this graph. */
     hasOp(name: string): boolean;
     /** Total number of ops in the graph. */

package/dist/graph.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../src/ts/graph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAA2B,MAAM,gBAAgB,CAAC;AAGhE,MAAM,MAAM,SAAS,GACjB;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAChC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,GAChC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,EAAE,CAAA;CAAE,GAClC;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,KAAK,EAAE,KAAK,EAAE,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,KAAK,EAAE,MAAM,EAAE,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,KAAK,EAAE,YAAY,CAAA;CAAE,CAAC;AAE5C,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,QAAQ,GAAG;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AAEzD;;;;;;;;;;;;GAYG;AACH,qBAAa,KAAK;IAChB,gBAAgB;IAChB,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC;gBAEV,MAAM,EAAE,GAAG;IAIvB;;;;;;;;OAQG;IACH,KAAK,CACH,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,QAAQ,EAAE,EAClB,KAAK,GAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAM,EACrC,IAAI,CAAC,EAAE,MAAM,GACZ,MAAM,EAAE;IAkCX,8DAA8D;IAC9D,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI5B,wCAAwC;IACxC,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED;;;OAGG;IACH,UAAU,IAAI,MAAM;IAIpB;;;;;;;;;;;;;;OAcG;IACH,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKpC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,SAAI,GAAG,MAAM,GAAG,IAAI;CAW9C"}
+{"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../src/ts/graph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAA2B,MAAM,gBAAgB,CAAC;AAGhE,MAAM,MAAM,SAAS,GACjB;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GAChC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,OAAO,CAAA;CAAE,GAChC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GAC9B;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,EAAE,CAAA;CAAE,GAClC;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,KAAK,EAAE,KAAK,EAAE,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,KAAK,EAAE,MAAM,EAAE,CAAA;CAAE,GACrC;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GACjC;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,KAAK,EAAE,YAAY,CAAA;CAAE,CAAC;AAE5C,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,KAAK,CAAC;IACb,KAAK,EAAE,MAAM,EAAE,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,MAAM,QAAQ,GAAG;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AAEzD;;;;;;;;;;;;GAYG;AACH,qBAAa,KAAK;IAChB,gBAAgB;IAChB,QAAQ,CAAC,OAAO,EAAE,GAAG,CAAC;gBAEV,MAAM,EAAE,GAAG;IAIvB;;;;;;;;;;;OAWG;IACH,KAAK,CACH,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,QAAQ,EAAE,EAClB,KAAK,GAAE,MAAM,CAAC,MAAM,EAAE,SAAS,CAAM,EACrC,IAAI,CAAC,EAAE,MAAM,EACb,aAAa,GAAE,MAAM,EAAO,GAC3B,MAAM,EAAE;IAwCX;;;;;;;;;;;;;;;;OAgBG;IACH,YAAY,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,EAAE,QAAQ,EAAE,GAAG,MAAM,EAAE;IAgBrE,8DAA8D;IAC9D,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAI5B,wCAAwC;IACxC,IAAI,MAAM,IAAI,MAAM,CAEnB;IAED;;;OAGG;IACH,UAAU,IAAI,MAAM;IAIpB;;;;;;;;;;;;;;OAcG;IACH,cAAc,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAKpC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,SAAI,GAAG,MAAM,GAAG,IAAI;CAW9C"}

package/dist/graph.js

@@ -25,9 +25,12 @@ export class Graph {
      * @param inputs    Output references from prior ops
      * @param attrs     Op attributes
      * @param name      Optional explicit op name (auto-generated if omitted)
+     * @param controlInputs Op names that must complete before this op runs.
+     *                      Used by globalVariablesInitializer to sequence init
+     *                      ops before the NoOp target that callers wait on.
      * @returns         Array of output Tensors (one per op output)
      */
-    addOp(type, inputs, attrs = {}, name) {
+    addOp(type, inputs, attrs = {}, name, controlInputs = []) {
         const nativeAttrs = {};
         for (const [k, v] of Object.entries(attrs)) {
             if (v.kind === "list_type") {
@@ -40,7 +43,7 @@ export class Graph {
                 nativeAttrs[k] = v;
             }
         }
-        const result = this._native.addOp(type, inputs, nativeAttrs, name);
+        const result = this._native.addOp(type, inputs, nativeAttrs, name, controlInputs);
         const { opName, numOutputs } = result;
         const tensors = [];
         for (let i = 0; i < numOutputs; i++) {
@@ -50,6 +53,31 @@ export class Graph {
         }
         return tensors;
     }
+    /**
+     * addGradients — compute symbolic gradients via TF_AddGradients.
+     *
+     * Injects gradient ops into the graph.  Returns one gradient Tensor per
+     * entry in `x` — the partial derivative dSum(y)/dx_i.
+     *
+     * @param y   Loss outputs to differentiate (typically a scalar loss tensor)
+     * @param x   Parameters to differentiate with respect to
+     * @param dx  Initial upstream gradients (default: ones, i.e. dL/dy = 1)
+     *
+     * @example
+     * const loss = ops.sparseSoftmaxCrossEntropyWithLogits(g, labels, logits);
+     * const wVal = ops.readVariable(g, wHandle, DType.FLOAT32);
+     * const [dw] = g.addGradients([loss], [wVal]);
+     * // dw is now a Tensor representing dLoss/dW
+     * // pass it to applyGradientDescent or applyAdam
+     */
+    addGradients(y, x, dx) {
+        const raw = this._native.addGradients(y, x, dx ?? null);
+        return raw.map(({ opName, index }) => {
+            const tfDtype = this._native.opOutputType(opName, index);
+            const tfShape = this._native.opOutputShape(opName, index);
+            return makeTensor(opName, index, tfDtype != null ? tfDtype : null, tfShape != null ? ShapeFromTF(tfShape) : null);
+        });
+    }
     /** Whether an op with the given name exists in this graph. */
     hasOp(name) {
         return this._native.hasOp(name);

package/dist/graph.js.map

@@ -1 +1 @@
-{"version":3,"file":"graph.js","sourceRoot":"","sources":["../src/ts/graph.ts"],"names":[],"mappings":"AACA,OAAO,EAAS,UAAU,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAqBhE;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,KAAK;IAChB,gBAAgB;IACP,OAAO,CAAM;IAEtB,YAAY,MAAW;QACrB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CACH,IAAY,EACZ,MAAkB,EAClB,QAAmC,EAAE,EACrC,IAAa;QAEb,MAAM,WAAW,GAAwB,EAAE,CAAC;QAC5C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC3B,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACrE,CAAC;iBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC7B,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,CAAC,CAAC;QACnE,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAG9B,CAAC;QAEF,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAkB,CAAC;YACtE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAoB,CAAC;YACzE,OAAO,CAAC,IAAI,CACV,UAAU,CACR,MAAM,EACN,CAAC,EACD,OAAO,IAAI,IAAI,CAAC,CAAC,CAAE,OAAiB,CAAC,CAAC,CAAC,IAAI,EAC3C,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAC9C,CACF,CAAC;QACJ,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,8DAA8D;IAC9D,KAAK,CAAC,IAAY;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAY,CAAC;IAC7C,CAAC;IAED,wCAAwC;IACxC,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,EAAY,CAAC;IACzC,CAAC;IAED;;;OAGG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAY,CAAC;IAC7C,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,cAAc,CAAC,MAAc;QAC3B,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;IAED,WAAW;IACX,KAAK,CAAC,IAAY,EAAE,KAAK,GAAG,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAkB,CAAC;QACxE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,KAAK,CAAoB,CAAC;QAC3E,OAAO,UAAU,CACf,IAAI,EACJ,KAAK,EACL,OAAO,IAAI,IAAI,CAAC,CAAC,CAAE,OAAiB,CAAC,CAAC,CAAC,IAAI,EAC3C,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAC9C,CAAC;IACJ,CAAC;CACF"}
+{"version":3,"file":"graph.js","sourceRoot":"","sources":["../src/ts/graph.ts"],"names":[],"mappings":"AACA,OAAO,EAAS,UAAU,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAsBhE;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,KAAK;IAChB,gBAAgB;IACP,OAAO,CAAM;IAEtB,YAAY,MAAW;QACrB,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;IACxB,CAAC;IAED;;;;;;;;;;;OAWG;IACH,KAAK,CACH,IAAY,EACZ,MAAkB,EAClB,QAAmC,EAAE,EACrC,IAAa,EACb,gBAA0B,EAAE;QAE5B,MAAM,WAAW,GAAwB,EAAE,CAAC;QAC5C,KAAK,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBAC3B,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACrE,CAAC;iBAAM,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAC7B,WAAW,CAAC,CAAC,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;YAC5D,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;QACH,CAAC;QAED,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAC/B,IAAI,EACJ,MAAM,EACN,WAAW,EACX,IAAI,EACJ,aAAa,CACd,CAAC;QACF,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,GAAG,MAG9B,CAAC;QAEF,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACpC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,CAAC,CAAkB,CAAC;YACtE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE,CAAC,CAAoB,CAAC;YACzE,OAAO,CAAC,IAAI,CACV,UAAU,CACR,MAAM,EACN,CAAC,EACD,OAAO,IAAI,IAAI,CAAC,CAAC,CAAE,OAAiB,CAAC,CAAC,CAAC,IAAI,EAC3C,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAC9C,CACF,CAAC;QACJ,CAAC;QACD,OAAO,OAAO,CAAC;IACjB,CAAC;IAED;;;;;;;;;;;;;;;;OAgBG;IACH,YAAY,CAAC,CAAa,EAAE,CAAa,EAAE,EAAe;QACxD,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,IAAI,IAAI,CAAe,CAAC;QACtE,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,EAAE,EAAE;YACnC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,MAAM,EAAE,KAAK,CAAkB,CAAC;YAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,MAAM,EAAE,KAAK,CAEhD,CAAC;YACT,OAAO,UAAU,CACf,MAAM,EACN,KAAK,EACL,OAAO,IAAI,IAAI,CAAC,CAAC,CAAE,OAAiB,CAAC,CAAC,CAAC,IAAI,EAC3C,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAC9C,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED,8DAA8D;IAC9D,KAAK,CAAC,IAAY;QAChB,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAY,CAAC;IAC7C,CAAC;IAED,wCAAwC;IACxC,IAAI,MAAM;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,EAAY,CAAC;IACzC,CAAC;IAED;;;OAGG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAY,CAAC;IAC7C,CAAC;IAED;;;;;;;;;;;;;;OAcG;IACH,cAAc,CAAC,MAAc;QAC3B,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IACtC,CAAC;IAED,WAAW;IACX,KAAK,CAAC,IAAY,EAAE,KAAK,GAAG,CAAC;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAC3C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,EAAE,KAAK,CAAkB,CAAC;QACxE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,KAAK,CAAoB,CAAC;QAC3E,OAAO,UAAU,CACf,IAAI,EACJ,KAAK,EACL,OAAO,IAAI,IAAI,CAAC,CAAC,CAAE,OAAiB,CAAC,CAAC,CAAC,IAAI,EAC3C,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAC9C,CAAC;IACJ,CAAC;CACF"}

package/dist/index.d.ts

@@ -44,4 +44,7 @@ export declare function session(g: Graph, options?: {
 export * as ops from "./ops/index.js";
 export type { PoolOptions, PoolResult, ExecutionStrategy, } from "./inference-pool.js";
 export { InferencePool } from "./inference-pool.js";
+export * as optimizers from "./optimizers/index.js";
+export type { ActivationFn, Layer, LayerParam, LossFn, TrainStepResult, } from "./model/index.js";
+export { Dense, Flatten, Conv2D, Sequential } from "./model/index.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/ts/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAsBH,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EACL,KAAK,EACL,aAAa,EACb,SAAS,EACT,UAAU,EACV,QAAQ,GACT,MAAM,gBAAgB,CAAC;AAGxB,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,YAAY,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3D,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAIvC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC;;;;;;GAMG;AACH,wBAAgB,KAAK,IAAI,KAAK,CAE7B;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,OAAO,CACrB,CAAC,EAAE,KAAK,EACR,OAAO,CAAC,EAAE;IACR,QAAQ,CAAC,EAAE,aAAa,GAAG,aAAa,CAAC;IACzC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GACA,OAAO,CAET;AAGD,OAAO,KAAK,GAAG,MAAM,gBAAgB,CAAC;AAGtC,YAAY,EACV,WAAW,EACX,UAAU,EACV,iBAAiB,GAClB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/ts/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAsBH,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EACL,KAAK,EACL,aAAa,EACb,SAAS,EACT,UAAU,EACV,QAAQ,GACT,MAAM,gBAAgB,CAAC;AAGxB,YAAY,EAAE,SAAS,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACpE,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,YAAY,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAC3D,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAIvC,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC;;;;;;GAMG;AACH,wBAAgB,KAAK,IAAI,KAAK,CAE7B;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAgB,OAAO,CACrB,CAAC,EAAE,KAAK,EACR,OAAO,CAAC,EAAE;IACR,QAAQ,CAAC,EAAE,aAAa,GAAG,aAAa,CAAC;IACzC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB,GACA,OAAO,CAET;AAGD,OAAO,KAAK,GAAG,MAAM,gBAAgB,CAAC;AAGtC,YAAY,EACV,WAAW,EACX,UAAU,EACV,iBAAiB,GAClB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAGpD,OAAO,KAAK,UAAU,MAAM,uBAAuB,CAAC;AACpD,YAAY,EACV,YAAY,EACZ,KAAK,EACL,UAAU,EACV,MAAM,EACN,eAAe,GAChB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC"}

package/dist/index.js

@@ -55,4 +55,7 @@ export function session(g, options) {
 // ── Ops namespace ─────────────────────────────────────────────────────────────
 export * as ops from "./ops/index.js";
 export { InferencePool } from "./inference-pool.js";
+// ── Model layers and optimizers ───────────────────────────────────────────────
+export * as optimizers from "./optimizers/index.js";
+export { Dense, Flatten, Conv2D, Sequential } from "./model/index.js";
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/ts/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AACtD,OAAO,YAAY,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAErC,gFAAgF;AAChF,yEAAyE;AACzE,qEAAqE;AACrE,sDAAsD;AACtD,MAAM,QAAQ,EAAE,CAAC;AAEjB,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,CAAQ,CAAC;AAE/D,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAIhB,OAAO,EACL,KAAK,EACL,aAAa,EACb,SAAS,EACT,UAAU,EACV,QAAQ,GACT,MAAM,gBAAgB,CAAC;AAIxB,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,iFAAiF;AAEjF,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC;;;;;;GAMG;AACH,MAAM,UAAU,KAAK;IACnB,OAAO,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;AACtC,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,OAAO,CACrB,CAAQ,EACR,OAKC;IAED,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,iFAAiF;AACjF,OAAO,KAAK,GAAG,MAAM,gBAAgB,CAAC;AAQtC,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/ts/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,4BAA4B,CAAC;AACtD,OAAO,YAAY,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAErC,gFAAgF;AAChF,yEAAyE;AACzE,qEAAqE;AACrE,sDAAsD;AACtD,MAAM,QAAQ,EAAE,CAAC;AAEjB,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,EAAE,IAAI,CAAC,CAAQ,CAAC;AAE/D,OAAO,EAAE,QAAQ,EAAE,MAAM,cAAc,CAAC;AACxC,QAAQ,CAAC,KAAK,CAAC,CAAC;AAIhB,OAAO,EACL,KAAK,EACL,aAAa,EACb,SAAS,EACT,UAAU,EACV,QAAQ,GACT,MAAM,gBAAgB,CAAC;AAIxB,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AAEnC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,iFAAiF;AAEjF,OAAO,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACnC,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC;;;;;;GAMG;AACH,MAAM,UAAU,KAAK;IACnB,OAAO,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC;AACtC,CAAC;AAED;;;;;;;;;;;;;GAaG;AACH,MAAM,UAAU,OAAO,CACrB,CAAQ,EACR,OAKC;IAED,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;AAC5D,CAAC;AAED,iFAAiF;AACjF,OAAO,KAAK,GAAG,MAAM,gBAAgB,CAAC;AAQtC,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,iFAAiF;AACjF,OAAO,KAAK,UAAU,MAAM,uBAAuB,CAAC;AAQpD,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC"}

package/dist/model/index.d.ts

@@ -0,0 +1,5 @@
+export type { ActivationFn, Layer, LayerParam } from "./layer.js";
+export { Dense, Flatten, Conv2D } from "./layers.js";
+export type { LossFn, TrainStepResult } from "./sequential.js";
+export { Sequential } from "./sequential.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/model/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ts/model/index.ts"],"names":[],"mappings":"AAAA,YAAY,EAAE,YAAY,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAClE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrD,YAAY,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/model/index.js

@@ -0,0 +1,3 @@
+export { Dense, Flatten, Conv2D } from "./layers.js";
+export { Sequential } from "./sequential.js";
+//# sourceMappingURL=index.js.map

package/dist/model/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/ts/model/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC"}

package/dist/model/layer.d.ts

@@ -0,0 +1,25 @@
+import type { Tensor } from "@isidorus/core";
+import { DType } from "@isidorus/core";
+import type { Graph } from "../graph.js";
+export type ActivationFn = "relu" | "leaky_relu" | "relu6" | "sigmoid" | "tanh" | "softmax" | "log_softmax" | "elu" | "selu" | "swish" | "gelu" | "linear";
+export interface LayerParam {
+    handle: Tensor;
+    read: Tensor;
+    dtype: DType;
+    name: string;
+    initOp: string;
+}
+export interface Layer {
+    readonly name: string;
+    /**
+     * Build — add ops to the graph.
+     * Called once by Sequential in order.
+     * @returns output shape after this layer (null dims = dynamic)
+     */
+    build(g: Graph, input: Tensor, inputShape: (number | null)[]): (number | null)[];
+    /** Symbolic output tensor. Set during build(). */
+    readonly output: Tensor;
+    /** Parameters exposed to the optimizer. Set during build(). */
+    readonly layerParams: LayerParam[];
+}
+//# sourceMappingURL=layer.d.ts.map

package/dist/model/layer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"layer.d.ts","sourceRoot":"","sources":["../../src/ts/model/layer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACvC,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AAkBzC,MAAM,MAAM,YAAY,GACpB,MAAM,GACN,YAAY,GACZ,OAAO,GACP,SAAS,GACT,MAAM,GACN,SAAS,GACT,aAAa,GACb,KAAK,GACL,MAAM,GACN,OAAO,GACP,MAAM,GACN,QAAQ,CAAC;AAEb,MAAM,WAAW,UAAU;IACzB,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,KAAK,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,KAAK;IACpB,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB;;;;OAIG;IACH,KAAK,CACH,CAAC,EAAE,KAAK,EACR,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,GAC5B,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,CAAC;IAErB,kDAAkD;IAClD,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IAExB,+DAA+D;IAC/D,QAAQ,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC;CACpC"}

package/dist/model/layer.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=layer.js.map

package/dist/model/layer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"layer.js","sourceRoot":"","sources":["../../src/ts/model/layer.ts"],"names":[],"mappings":""}

package/dist/model/layers.d.ts

@@ -0,0 +1,47 @@
+import type { Tensor } from "@isidorus/core";
+import type { Graph } from "../graph.js";
+import type { Layer, LayerParam, ActivationFn } from "./layer.js";
+export declare class Dense implements Layer {
+    readonly name: string;
+    output: Tensor;
+    readonly layerParams: LayerParam[];
+    private readonly units;
+    private readonly activation;
+    private readonly useBias;
+    constructor(units: number, options?: {
+        activation?: ActivationFn;
+        useBias?: boolean;
+        name?: string;
+    });
+    build(g: Graph, input: Tensor, inputShape: (number | null)[]): (number | null)[];
+}
+export declare class Flatten implements Layer {
+    readonly name: string;
+    output: Tensor;
+    readonly layerParams: LayerParam[];
+    constructor(options?: {
+        name?: string;
+    });
+    build(g: Graph, input: Tensor, inputShape: (number | null)[]): (number | null)[];
+}
+export declare class Conv2D implements Layer {
+    readonly name: string;
+    output: Tensor;
+    readonly layerParams: LayerParam[];
+    private readonly filters;
+    private readonly kernelSize;
+    private readonly strides;
+    private readonly padding;
+    private readonly activation;
+    private readonly useBias;
+    constructor(filters: number, options?: {
+        kernelSize?: number | [number, number];
+        strides?: number | [number, number];
+        padding?: "SAME" | "VALID";
+        activation?: ActivationFn;
+        useBias?: boolean;
+        name?: string;
+    });
+    build(g: Graph, input: Tensor, inputShape: (number | null)[]): (number | null)[];
+}
+//# sourceMappingURL=layers.d.ts.map

package/dist/model/layers.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"layers.d.ts","sourceRoot":"","sources":["../../src/ts/model/layers.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAE7C,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AA8DlE,qBAAa,KAAM,YAAW,KAAK;IACjC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,MAAM,EAAG,MAAM,CAAC;IAChB,QAAQ,CAAC,WAAW,EAAE,UAAU,EAAE,CAAM;IAExC,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAS;IAC/B,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAe;IAC1C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAU;gBAGhC,KAAK,EAAE,MAAM,EACb,OAAO,GAAE;QACP,UAAU,CAAC,EAAE,YAAY,CAAC;QAC1B,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,IAAI,CAAC,EAAE,MAAM,CAAC;KACV;IAQR,KAAK,CACH,CAAC,EAAE,KAAK,EACR,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,GAC5B,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE;CAoErB;AAKD,qBAAa,OAAQ,YAAW,KAAK;IACnC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,MAAM,EAAG,MAAM,CAAC;IAChB,QAAQ,CAAC,WAAW,EAAE,UAAU,EAAE,CAAM;gBAE5B,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAA;KAAO;IAI3C,KAAK,CACH,CAAC,EAAE,KAAK,EACR,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,GAC5B,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE;CA2BrB;AAKD,qBAAa,MAAO,YAAW,KAAK;IAClC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,MAAM,EAAG,MAAM,CAAC;IAChB,QAAQ,CAAC,WAAW,EAAE,UAAU,EAAE,CAAM;IAExC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAmB;IAC9C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmC;IAC3D,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAmB;IAC3C,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAe;IAC1C,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAU;gBAGhC,OAAO,EAAE,MAAM,EACf,OAAO,GAAE;QACP,UAAU,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,OAAO,CAAC,EAAE,MAAM,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpC,OAAO,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC;QAC3B,UAAU,CAAC,EAAE,YAAY,CAAC;QAC1B,OAAO,CAAC,EAAE,OAAO,CAAC;QAClB,IAAI,CAAC,EAAE,MAAM,CAAC;KACV;IAgBR,KAAK,CACH,CAAC,EAAE,KAAK,EACR,KAAK,EAAE,MAAM,EACb,UAAU,EAAE,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE,GAC5B,CAAC,MAAM,GAAG,IAAI,CAAC,EAAE;CAqGrB"}

package/dist/model/layers.js

@@ -0,0 +1,191 @@
+import { DType } from "@isidorus/core";
+import { variableWithInit, readVariable, zerosInitializer, glorotUniformInitializer, truncatedNormalInitializer, } from "../ops/variable_ops.js";
+import { constant } from "../ops/array_ops.js";
+import { matmul, biasAdd } from "../ops/math_ops.js";
+import { relu, leakyRelu, relu6, sigmoid, conv2d as conv2dOp, tanh, softmax, logSoftmax, elu, selu, swish, gelu, } from "../ops/nn_ops.js";
+// ---------------------------------------------------------------------------
+// Activation helper
+// ---------------------------------------------------------------------------
+function activate(g, x, fn, name) {
+    switch (fn) {
+        case "relu":
+            return relu(g, x, `${name}/relu`);
+        case "leaky_relu":
+            return leakyRelu(g, x, 0.2, `${name}/leaky_relu`);
+        case "relu6":
+            return relu6(g, x, `${name}/relu6`);
+        case "sigmoid":
+            return sigmoid(g, x, `${name}/sigmoid`);
+        case "tanh":
+            return tanh(g, x, `${name}/tanh`);
+        case "softmax":
+            return softmax(g, x, `${name}/softmax`);
+        case "log_softmax":
+            return logSoftmax(g, x, `${name}/log_softmax`);
+        case "elu":
+            return elu(g, x, `${name}/elu`);
+        case "selu":
+            return selu(g, x, `${name}/selu`);
+        case "swish":
+            return swish(g, x, `${name}/swish`);
+        case "gelu":
+            return gelu(g, x, `${name}/gelu`);
+        case "linear":
+            return x;
+        default:
+            throw new Error(`Unknown activation: ${fn}`);
+    }
+}
+// ---------------------------------------------------------------------------
+// Dense — fully-connected layer: output = activation(input @ W + b)
+// ---------------------------------------------------------------------------
+export class Dense {
+    name;
+    output;
+    layerParams = [];
+    units;
+    activation;
+    useBias;
+    constructor(units, options = {}) {
+        this.units = units;
+        this.activation = options.activation ?? "linear";
+        this.useBias = options.useBias ?? true;
+        this.name = options.name ?? `dense_${units}`;
+    }
+    build(g, input, inputShape) {
+        const inFeatures = inputShape[inputShape.length - 1];
+        if (!inFeatures || inFeatures < 1)
+            throw new Error(`Dense "${this.name}": last input dim must be known, got ${JSON.stringify(inputShape)}`);
+        // ── Weight W: [inFeatures, units] ────────────────────────────────────
+        const wInitVal = glorotUniformInitializer(g, [inFeatures, this.units], DType.FLOAT32, `${this.name}/w_glorot`);
+        const { handle: wHandle, initOp: wInitOp } = variableWithInit(g, [inFeatures, this.units], DType.FLOAT32, `${this.name}/w`, wInitVal);
+        const wRead = readVariable(g, wHandle, DType.FLOAT32, `${this.name}/w_read`);
+        let out = matmul(g, input, wRead, {}, `${this.name}/matmul`);
+        this.layerParams.push({
+            handle: wHandle,
+            read: wRead,
+            dtype: DType.FLOAT32,
+            name: `${this.name}/w`,
+            initOp: wInitOp,
+        });
+        // ── Bias b: [units] ───────────────────────────────────────────────────
+        if (this.useBias) {
+            const bInitVal = zerosInitializer(g, [this.units], DType.FLOAT32);
+            const { handle: bHandle, initOp: bInitOp } = variableWithInit(g, [this.units], DType.FLOAT32, `${this.name}/b`, bInitVal);
+            const bRead = readVariable(g, bHandle, DType.FLOAT32, `${this.name}/b_read`);
+            out = biasAdd(g, out, bRead, `${this.name}/bias_add`);
+            this.layerParams.push({
+                handle: bHandle,
+                read: bRead,
+                dtype: DType.FLOAT32,
+                name: `${this.name}/b`,
+                initOp: bInitOp,
+            });
+        }
+        this.output = activate(g, out, this.activation, this.name);
+        return [...inputShape.slice(0, -1), this.units];
+    }
+}
+// ---------------------------------------------------------------------------
+// Flatten — reshapes [batch, d1, d2, ...] → [batch, d1*d2*...]
+// ---------------------------------------------------------------------------
+export class Flatten {
+    name;
+    output;
+    layerParams = [];
+    constructor(options = {}) {
+        this.name = options.name ?? "flatten";
+    }
+    build(g, input, inputShape) {
+        const spatialDims = inputShape.slice(1);
+        const hasUnknown = spatialDims.some((d) => d === null);
+        const flatSize = hasUnknown
+            ? null
+            : spatialDims.reduce((a, b) => a * b, 1);
+        const flatDim = flatSize ?? -1;
+        const shapeBuf = Buffer.allocUnsafe(8);
+        shapeBuf.writeInt32LE(-1, 0);
+        shapeBuf.writeInt32LE(flatDim, 4);
+        const shapeConst = constant(g, shapeBuf, [2], DType.INT32, `${this.name}/shape`);
+        const [out] = g.addOp("Reshape", [input, shapeConst], {}, `${this.name}/reshape`);
+        this.output = out;
+        return [null, flatSize];
+    }
+}
+// ---------------------------------------------------------------------------
+// Conv2D — 2D convolution (NHWC): output = activation(conv2d(input, W) + b)
+// ---------------------------------------------------------------------------
+export class Conv2D {
+    name;
+    output;
+    layerParams = [];
+    filters;
+    kernelSize;
+    strides;
+    padding;
+    activation;
+    useBias;
+    constructor(filters, options = {}) {
+        this.filters = filters;
+        this.padding = options.padding ?? "SAME";
+        this.activation = options.activation ?? "linear";
+        this.useBias = options.useBias ?? true;
+        this.name = options.name ?? `conv2d_${filters}f`;
+        const ks = options.kernelSize ?? 3;
+        this.kernelSize = Array.isArray(ks) ? ks : [ks, ks];
+        const st = options.strides ?? 1;
+        const [sH, sW] = Array.isArray(st) ? st : [st, st];
+        this.strides = [1, sH, sW, 1];
+    }
+    build(g, input, inputShape) {
+        if (inputShape.length !== 4)
+            throw new Error(`Conv2D "${this.name}": expects 4D input [batch,H,W,C], got rank ${inputShape.length}`);
+        const inChannels = inputShape[3];
+        if (!inChannels || inChannels < 1)
+            throw new Error(`Conv2D "${this.name}": in_channels must be known, got ${JSON.stringify(inputShape)}`);
+        const [kH, kW] = this.kernelSize;
+        const wShape = [kH, kW, inChannels, this.filters];
+        // He normal init for conv: stddev = sqrt(2 / (kH * kW * inChannels))
+        const stddev = Math.sqrt(2 / (kH * kW * inChannels));
+        const wInitVal = truncatedNormalInitializer(g, wShape, DType.FLOAT32, { stddev }, `${this.name}/w_init`);
+        const { handle: wHandle, initOp: wInitOp } = variableWithInit(g, wShape, DType.FLOAT32, `${this.name}/w`, wInitVal);
+        const wRead = readVariable(g, wHandle, DType.FLOAT32, `${this.name}/w_read`);
+        let out = conv2dOp(g, input, wRead, this.strides, this.padding, `${this.name}/conv`);
+        this.layerParams.push({
+            handle: wHandle,
+            read: wRead,
+            dtype: DType.FLOAT32,
+            name: `${this.name}/w`,
+            initOp: wInitOp,
+        });
+        if (this.useBias) {
+            const bInitVal = zerosInitializer(g, [this.filters], DType.FLOAT32);
+            const { handle: bHandle, initOp: bInitOp } = variableWithInit(g, [this.filters], DType.FLOAT32, `${this.name}/b`, bInitVal);
+            const bRead = readVariable(g, bHandle, DType.FLOAT32, `${this.name}/b_read`);
+            out = biasAdd(g, out, bRead, `${this.name}/bias_add`);
+            this.layerParams.push({
+                handle: bHandle,
+                read: bRead,
+                dtype: DType.FLOAT32,
+                name: `${this.name}/b`,
+                initOp: bInitOp,
+            });
+        }
+        this.output = activate(g, out, this.activation, this.name);
+        // Output spatial dimensions
+        const [, H, W] = inputShape;
+        const [, sH, sW] = this.strides;
+        const outH = H === null
+            ? null
+            : this.padding === "SAME"
+                ? Math.ceil(H / sH)
+                : Math.ceil((H - kH + 1) / sH);
+        const outW = W === null
+            ? null
+            : this.padding === "SAME"
+                ? Math.ceil(W / sW)
+                : Math.ceil((W - kW + 1) / sW);
+        return [null, outH, outW, this.filters];
+    }
+}
+//# sourceMappingURL=layers.js.map

package/dist/model/layers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"layers.js","sourceRoot":"","sources":["../../src/ts/model/layers.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AAGvC,OAAO,EACL,gBAAgB,EAChB,YAAY,EACZ,gBAAgB,EAChB,wBAAwB,EACxB,0BAA0B,GAC3B,MAAM,wBAAwB,CAAC;AAChC,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAC/C,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EACL,IAAI,EACJ,SAAS,EACT,KAAK,EACL,OAAO,EACP,MAAM,IAAI,QAAQ,EAClB,IAAI,EACJ,OAAO,EACP,UAAU,EACV,GAAG,EACH,IAAI,EACJ,KAAK,EACL,IAAI,GACL,MAAM,kBAAkB,CAAC;AAE1B,8EAA8E;AAC9E,oBAAoB;AACpB,8EAA8E;AAC9E,SAAS,QAAQ,CAAC,CAAQ,EAAE,CAAS,EAAE,EAAgB,EAAE,IAAY;IACnE,QAAQ,EAAE,EAAE,CAAC;QACX,KAAK,MAAM;YACT,OAAO,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,OAAO,CAAC,CAAC;QACpC,KAAK,YAAY;YACf,OAAO,SAAS,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,IAAI,aAAa,CAAC,CAAC;QACpD,KAAK,OAAO;YACV,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,CAAC;QACtC,KAAK,SAAS;YACZ,OAAO,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,UAAU,CAAC,CAAC;QAC1C,KAAK,MAAM;YACT,OAAO,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,OAAO,CAAC,CAAC;QACpC,KAAK,SAAS;YACZ,OAAO,OAAO,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,UAAU,CAAC,CAAC;QAC1C,KAAK,aAAa;YAChB,OAAO,UAAU,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,cAAc,CAAC,CAAC;QACjD,KAAK,KAAK;YACR,OAAO,GAAG,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,MAAM,CAAC,CAAC;QAClC,KAAK,MAAM;YACT,OAAO,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,OAAO,CAAC,CAAC;QACpC,KAAK,OAAO;YACV,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,QAAQ,CAAC,CAAC;QACtC,KAAK,MAAM;YACT,OAAO,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,IAAI,OAAO,CAAC,CAAC;QACpC,KAAK,QAAQ;YACX,OAAO,CAAC,CAAC;QACX;YACE,MAAM,IAAI,KAAK,CAAC,uBAAuB,EAAE,EAAE,CAAC,CAAC;IACjD,CAAC;AACH,CAAC;AAED,8EAA8E;AAC9E,oEAAoE;AACpE,8EAA8E;AAC9E,MAAM,OAAO,KAAK;IACP,IAAI,CAAS;IACtB,MAAM,CAAU;IACP,WAAW,GAAiB,EAAE,CAAC;IAEvB,KAAK,CAAS;IACd,UAAU,CAAe;IACzB,OAAO,CAAU;IAElC,YACE,KAAa,EACb,UAII,EAAE;QAEN,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,QAAQ,CAAC;QACjD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,SAAS,KAAK,EAAE,CAAC;IAC/C,CAAC;IAED,KAAK,CACH,CAAQ,EACR,KAAa,EACb,UAA6B;QAE7B,MAAM,UAAU,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAW,CAAC;QAC/D,IAAI,CAAC,UAAU,IAAI,UAAU,GAAG,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,UACE,IAAI,CAAC,IACP,wCAAwC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,EAAE,CACrE,CAAC;QAEJ,wEAAwE;QACxE,MAAM,QAAQ,GAAG,wBAAwB,CACvC,CAAC,EACD,CAAC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,EACxB,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,WAAW,CACxB,CAAC;QACF,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,gBAAgB,CAC3D,CAAC,EACD,CAAC,UAAU,EAAE,IAAI,CAAC,KAAK,CAAC,EACxB,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,IAAI,EAChB,QAAQ,CACT,CAAC;QACF,MAAM,KAAK,GAAG,YAAY,CACxB,CAAC,EACD,OAAO,EACP,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,SAAS,CACtB,CAAC;QAEF,IAAI,GAAG,GAAG,MAAM,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,GAAG,IAAI,CAAC,IAAI,SAAS,CAAC,CAAC;QAC7D,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YACpB,MAAM,EAAE,OAAO;YACf,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK,CAAC,OAAO;YACpB,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI;YACtB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;QAEH,yEAAyE;QACzE,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YAClE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,gBAAgB,CAC3D,CAAC,EACD,CAAC,IAAI,CAAC,KAAK,CAAC,EACZ,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,IAAI,EAChB,QAAQ,CACT,CAAC;YACF,MAAM,KAAK,GAAG,YAAY,CACxB,CAAC,EACD,OAAO,EACP,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,SAAS,CACtB,CAAC;YACF,GAAG,GAAG,OAAO,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,IAAI,WAAW,CAAC,CAAC;YACtD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;gBACpB,MAAM,EAAE,OAAO;gBACf,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,KAAK,CAAC,OAAO;gBACpB,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI;gBACtB,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAC3D,OAAO,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;IAClD,CAAC;CACF;AAED,8EAA8E;AAC9E,+DAA+D;AAC/D,8EAA8E;AAC9E,MAAM,OAAO,OAAO;IACT,IAAI,CAAS;IACtB,MAAM,CAAU;IACP,WAAW,GAAiB,EAAE,CAAC;IAExC,YAAY,UAA6B,EAAE;QACzC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,SAAS,CAAC;IACxC,CAAC;IAED,KAAK,CACH,CAAQ,EACR,KAAa,EACb,UAA6B;QAE7B,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,UAAU,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;QACvD,MAAM,QAAQ,GAAG,UAAU;YACzB,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAE,GAAG,CAAE,EAAE,CAAC,CAAC,CAAC;QAC7C,MAAM,OAAO,GAAG,QAAQ,IAAI,CAAC,CAAC,CAAC;QAE/B,MAAM,QAAQ,GAAG,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;QACvC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7B,QAAQ,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,UAAU,GAAG,QAAQ,CACzB,CAAC,EACD,QAAQ,EACR,CAAC,CAAC,CAAC,EACH,KAAK,CAAC,KAAK,EACX,GAAG,IAAI,CAAC,IAAI,QAAQ,CACrB,CAAC;QACF,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,KAAK,CACnB,SAAS,EACT,CAAC,KAAK,EAAE,UAAU,CAAC,EACnB,EAAE,EACF,GAAG,IAAI,CAAC,IAAI,UAAU,CACvB,CAAC;QACF,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;QAClB,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC1B,CAAC;CACF;AAED,8EAA8E;AAC9E,4EAA4E;AAC5E,8EAA8E;AAC9E,MAAM,OAAO,MAAM;IACR,IAAI,CAAS;IACtB,MAAM,CAAU;IACP,WAAW,GAAiB,EAAE,CAAC;IAEvB,OAAO,CAAS;IAChB,UAAU,CAAmB;IAC7B,OAAO,CAAmC;IAC1C,OAAO,CAAmB;IAC1B,UAAU,CAAe;IACzB,OAAO,CAAU;IAElC,YACE,OAAe,EACf,UAOI,EAAE;QAEN,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,MAAM,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,IAAI,QAAQ,CAAC;QACjD,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,IAAI,IAAI,CAAC;QACvC,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,IAAI,UAAU,OAAO,GAAG,CAAC;QAEjD,MAAM,EAAE,GAAG,OAAO,CAAC,UAAU,IAAI,CAAC,CAAC;QACnC,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,EAAuB,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QAE1E,MAAM,EAAE,GAAG,OAAO,CAAC,OAAO,IAAI,CAAC,CAAC;QAChC,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,EAAuB,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;QACzE,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAChC,CAAC;IAED,KAAK,CACH,CAAQ,EACR,KAAa,EACb,UAA6B;QAE7B,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC;YACzB,MAAM,IAAI,KAAK,CACb,WAAW,IAAI,CAAC,IAAI,+CAA+C,UAAU,CAAC,MAAM,EAAE,CACvF,CAAC;QAEJ,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAW,CAAC;QAC3C,IAAI,CAAC,UAAU,IAAI,UAAU,GAAG,CAAC;YAC/B,MAAM,IAAI,KAAK,CACb,WAAW,IAAI,CAAC,IAAI,qCAAqC,IAAI,CAAC,SAAS,CACrE,UAAU,CACX,EAAE,CACJ,CAAC;QAEJ,MAAM,CAAC,EAAE,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC;QACjC,MAAM,MAAM,GAAG,CAAC,EAAE,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAElD,qEAAqE;QACrE,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,0BAA0B,CACzC,CAAC,EACD,MAAM,EACN,KAAK,CAAC,OAAO,EACb,EAAE,MAAM,EAAE,EACV,GAAG,IAAI,CAAC,IAAI,SAAS,CACtB,CAAC;QACF,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,gBAAgB,CAC3D,CAAC,EACD,MAAM,EACN,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,IAAI,EAChB,QAAQ,CACT,CAAC;QACF,MAAM,KAAK,GAAG,YAAY,CACxB,CAAC,EACD,OAAO,EACP,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,SAAS,CACtB,CAAC;QAEF,IAAI,GAAG,GAAG,QAAQ,CAChB,CAAC,EACD,KAAK,EACL,KAAK,EACL,IAAI,CAAC,OAAO,EACZ,IAAI,CAAC,OAAO,EACZ,GAAG,IAAI,CAAC,IAAI,OAAO,CACpB,CAAC;QACF,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;YACpB,MAAM,EAAE,OAAO;YACf,IAAI,EAAE,KAAK;YACX,KAAK,EAAE,KAAK,CAAC,OAAO;YACpB,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI;YACtB,MAAM,EAAE,OAAO;SAChB,CAAC,CAAC;QAEH,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACjB,MAAM,QAAQ,GAAG,gBAAgB,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;YACpE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,gBAAgB,CAC3D,CAAC,EACD,CAAC,IAAI,CAAC,OAAO,CAAC,EACd,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,IAAI,EAChB,QAAQ,CACT,CAAC;YACF,MAAM,KAAK,GAAG,YAAY,CACxB,CAAC,EACD,OAAO,EACP,KAAK,CAAC,OAAO,EACb,GAAG,IAAI,CAAC,IAAI,SAAS,CACtB,CAAC;YACF,GAAG,GAAG,OAAO,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,GAAG,IAAI,CAAC,IAAI,WAAW,CAAC,CAAC;YACtD,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC;gBACpB,MAAM,EAAE,OAAO;gBACf,IAAI,EAAE,KAAK;gBACX,KAAK,EAAE,KAAK,CAAC,OAAO;gBACpB,IAAI,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI;gBACtB,MAAM,EAAE,OAAO;aAChB,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,QAAQ,CAAC,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;QAE3D,4BAA4B;QAC5B,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,GAAG,UAAU,CAAC;QAC5B,MAAM,CAAC,EAAE,EAAE,EAAE,EAAE,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC;QAChC,MAAM,IAAI,GACR,CAAC,KAAK,IAAI;YACR,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,IAAI,CAAC,OAAO,KAAK,MAAM;gBACzB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC;gBACnB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;QACnC,MAAM,IAAI,GACR,CAAC,KAAK,IAAI;YACR,CAAC,CAAC,IAAI;YACN,CAAC,CAAC,IAAI,CAAC,OAAO,KAAK,MAAM;gBACzB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,CAAC;gBACnB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC;QAEnC,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1C,CAAC;CACF"}

package/dist/model/sequential.d.ts

@@ -0,0 +1,91 @@
+import type { Tensor } from "@isidorus/core";
+import { DType } from "@isidorus/core";
+import type { Graph } from "../graph.js";
+import type { Session } from "../session.js";
+import type { Layer } from "./layer.js";
+import type { ParamSpec, FeedEntry } from "../optimizers/sgd.js";
+export type LossFn = "sparse_categorical_crossentropy" | "binary_crossentropy" | "mse";
+export interface Optimizer {
+    init(sess: Session): Promise<void>;
+    applyGradients(sess: Session, feeds: FeedEntry[]): Promise<void>;
+}
+export interface TrainStepResult {
+    loss: number;
+}
+export declare class Sequential {
+    private readonly g;
+    private readonly layers;
+    private _xPlaceholder;
+    private _yPlaceholder;
+    private _outputTensor;
+    private _lossTensor;
+    private _allParams;
+    private _allInitOp;
+    private _labelDtype;
+    private compiled;
+    constructor(g: Graph, layers: Layer[]);
+    /**
+     * compile — wire the full computation graph.
+     *
+     * After this returns, model.params contains ParamSpec entries with
+     * real gradient tensors from g.addGradients(). Use those to construct
+     * your optimizer before calling init().
+     */
+    compile(opts: {
+        loss: LossFn;
+        inputShape: number[];
+        labelDtype?: DType;
+    }): void;
+    /**
+     * init — run all variable initialisations and the optimizer's state init.
+     *
+     * @param sess The session to run on
+     * @param opt  The optimizer (must be constructed from model.params)
+     */
+    init(sess: Session, opt: Optimizer): Promise<void>;
+    /**
+     * trainStep — one forward pass, gradient computation, and weight update.
+     *
+     * Two sequential TF_SessionRun calls:
+     *   1. Fetch loss (forward pass runs as part of this)
+     *   2. Run optimizer update ops (backward pass + weight update)
+     *
+     * Keeping them separate avoids the optimizer needing to expose its
+     * internal step op name. The overhead is one extra C++ call per step,
+     * which is negligible compared to the matmul/conv cost.
+     *
+     * @param sess       Session to run on
+     * @param opt        Optimizer (same instance used in init)
+     * @param xBuf       Float32 input bytes [batchSize, ...inputShape]
+     * @param yBuf       Label bytes (INT32 class indices or FLOAT32 values)
+     * @param xShape     [batchSize, ...inputShape]
+     * @param yShape     [batchSize] for classification, [batchSize, units] for mse
+     * @param labelDtype DType for labels — defaults to what was set in compile()
+     */
+    trainStep(sess: Session, opt: Optimizer, xBuf: Buffer, yBuf: Buffer, xShape: number[], yShape: number[], labelDtype?: DType): Promise<TrainStepResult>;
+    /**
+     * predict — forward pass only, no gradient computation or update.
+     */
+    predict(sess: Session, xBuf: Buffer, xShape: number[]): Promise<{
+        data: Buffer;
+        shape: number[];
+        dtype: DType;
+    }>;
+    /** Input placeholder tensor — available after compile(). */
+    get xPlaceholder(): Tensor;
+    /** Label placeholder tensor — available after compile(). */
+    get yPlaceholder(): Tensor;
+    /** Final layer output tensor — available after compile(). */
+    get output(): Tensor;
+    /** Scalar mean loss tensor — available after compile(). */
+    get loss(): Tensor;
+    /**
+     * All parameter specs with real gradient tensors.
+     * Available after compile(). Use these to construct your optimizer.
+     */
+    get params(): ParamSpec[];
+    /** Label dtype resolved during compile(). */
+    get labelDtype(): DType;
+    private assertCompiled;
+}
+//# sourceMappingURL=sequential.d.ts.map

package/dist/model/sequential.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sequential.d.ts","sourceRoot":"","sources":["../../src/ts/model/sequential.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,OAAO,EAAE,KAAK,EAAE,MAAM,gBAAgB,CAAC;AACvC,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAC7C,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AASjE,MAAM,MAAM,MAAM,GACd,iCAAiC,GACjC,qBAAqB,GACrB,KAAK,CAAC;AAKV,MAAM,WAAW,SAAS;IACxB,IAAI,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;IACnC,cAAc,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;CAClE;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,MAAM,CAAC;CACd;AAqCD,qBAAa,UAAU;IACrB,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAQ;IAC1B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAU;IAGjC,OAAO,CAAC,aAAa,CAAU;IAC/B,OAAO,CAAC,aAAa,CAAU;IAC/B,OAAO,CAAC,aAAa,CAAU;IAC/B,OAAO,CAAC,WAAW,CAAU;IAC7B,OAAO,CAAC,UAAU,CAAe;IACjC,OAAO,CAAC,UAAU,CAAU;IAC5B,OAAO,CAAC,WAAW,CAAS;IAC5B,OAAO,CAAC,QAAQ,CAAS;gBAEb,CAAC,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE;IAKrC;;;;;;OAMG;IACH,OAAO,CAAC,IAAI,EAAE;QACZ,IAAI,EAAE,MAAM,CAAC;QACb,UAAU,EAAE,MAAM,EAAE,CAAC;QACrB,UAAU,CAAC,EAAE,KAAK,CAAC;KACpB,GAAG,IAAI;IAmJR;;;;;OAKG;IACG,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC;IAMxD;;;;;;;;;;;;;;;;;;OAkBG;IACG,SAAS,CACb,IAAI,EAAE,OAAO,EACb,GAAG,EAAE,SAAS,EACd,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,MAAM,EAAE,EAChB,MAAM,EAAE,MAAM,EAAE,EAChB,UAAU,CAAC,EAAE,KAAK,GACjB,OAAO,CAAC,eAAe,CAAC;IAuB3B;;OAEG;IACG,OAAO,CACX,IAAI,EAAE,OAAO,EACb,IAAI,EAAE,MAAM,EACZ,MAAM,EAAE,MAAM,EAAE,GACf,OAAO,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,EAAE,CAAC;QAAC,KAAK,EAAE,KAAK,CAAA;KAAE,CAAC;IAgB3D,4DAA4D;IAC5D,IAAI,YAAY,IAAI,MAAM,CAGzB;IAED,4DAA4D;IAC5D,IAAI,YAAY,IAAI,MAAM,CAGzB;IAED,6DAA6D;IAC7D,IAAI,MAAM,IAAI,MAAM,CAGnB;IAED,2DAA2D;IAC3D,IAAI,IAAI,IAAI,MAAM,CAGjB;IAED;;;OAGG;IACH,IAAI,MAAM,IAAI,SAAS,EAAE,CAGxB;IAED,6CAA6C;IAC7C,IAAI,UAAU,IAAI,KAAK,CAGtB;IAED,OAAO,CAAC,cAAc;CAIvB"}