tensorgrad 0.0.11 → 0.0.13

package/src/trace.ts

@@ -84,20 +84,25 @@ export function traceInto<T>(g: Graph, fn: () => T): T {
 // Their .source on the Tensor points at that node so codegen knows where to
 // bind external data.
 
+// Param/tensor inputs share a namespace (a step() call passes both as keys in
+// the same dispatch object); state inputs have their own namespace.
+type NamedInputKind = 'param_input' | 'tensor_input' | 'state_input'
+function assertNameUnused(g: Graph, name: string, kinds: NamedInputKind[], label: string): void {
+  if (g.ops.some(op => kinds.includes(op.kind as NamedInputKind) && (op as { name?: string }).name === name)) {
+    throw new Error(`tensorgrad: ${label} name '${name}' already used in this trace`)
+  }
+}
+
 export function paramInput(name: string, shape: Shape, dtype: Dtype = 'f32'): Tensor {
   const g = currentGraph()
-  if (g.ops.some(op => (op.kind === 'param_input' || op.kind === 'tensor_input') && op.name === name)) {
-    throw new Error(`tensorgrad: input name '${name}' already used in this trace`)
-  }
+  assertNameUnused(g, name, ['param_input', 'tensor_input'], 'input')
   const site = captureSite('paramInput')
   return addOp(g, 'param_input', shape, dtype, site, { name } as any)
 }
 
 export function tensorInput(name: string, shape: Shape, dtype: Dtype = 'f32'): Tensor {
   const g = currentGraph()
-  if (g.ops.some(op => (op.kind === 'param_input' || op.kind === 'tensor_input') && op.name === name)) {
-    throw new Error(`tensorgrad: input name '${name}' already used in this trace`)
-  }
+  assertNameUnused(g, name, ['param_input', 'tensor_input'], 'input')
   const site = captureSite('tensorInput')
   return addOp(g, 'tensor_input', shape, dtype, site, { name } as any)
 }
@@ -106,9 +111,7 @@ export function tensorInput(name: string, shape: Shape, dtype: Dtype = 'f32'): T
 // and updated across step() calls via writebacks declared by the optimizer helper.
 export function stateInput(name: string, shape: Shape, dtype: Dtype = 'f32', initValue = 0): Tensor {
   const g = currentGraph()
-  if (g.ops.some(op => op.kind === 'state_input' && op.name === name)) {
-    throw new Error(`tensorgrad: state name '${name}' already used in this trace`)
-  }
+  assertNameUnused(g, name, ['state_input'], 'state')
   const site = captureSite('stateInput')
   return addOp(g, 'state_input', shape, dtype, site, { name, initValue } as any)
 }

package/dist/adam.d.ts

@@ -1,65 +0,0 @@
-import type { Tensor } from './ir.js';
-import type { Graph } from './ir.js';
-import type { WritebackDecl } from './buffers.js';
-export interface AdamConfig {
-    /** Constant scalar (e.g., `0.005`) or a per-step schedule function
-     *  `(step) => lr`. Schedule fn lets the user implement linear/cosine decay
-     *  or warmup; first call passes `step=1`. Decay-shrink (AdamW) updates
-     *  per-step automatically when this is a function. */
-    lr: number | ((step: number) => number);
-    b1?: number;
-    b2?: number;
-    eps?: number;
-    /** AdamW: decoupled weight decay coefficient. Default 0 (plain Adam).
-     *  When non-zero, every step shrinks each decayed param by a factor of
-     *  `1 - lr * weightDecay` before the gradient update. */
-    weightDecay?: number;
-    /** Filter deciding which params get weight decay. Only consulted when
-     *  weightDecay > 0. Default: decay every param. Override for the standard
-     *  transformer convention (decay weights/embeddings, skip biases + LN gains).
-     *  Example: `(name) => name.includes('.W') || name.endsWith('_emb')`. */
-    decayFilter?: (paramName: string) => boolean;
-}
-/** Resolved hyperparameters: lr is the schedule fn (constants are wrapped). */
-export interface AdamResolvedConfig {
-    lr: (step: number) => number;
-    b1: number;
-    b2: number;
-    eps: number;
-    weightDecay: number;
-    decayFilter: (name: string) => boolean;
-    /** True iff the user supplied an lr function (vs a constant). When false,
-     *  decayShrink is baked at compile time and never updated. */
-    lrIsScheduled: boolean;
-}
-export interface AdamResult {
-    /** Writebacks the buffer planner should wire into the runtime. */
-    writebacks: WritebackDecl[];
-    /** Name of the per-step scalar tensor_input. The runtime fills this each call
-     * with `lr * sqrt(1-b2^t)/(1-b1^t)` (Adam's bias-corrected effective LR). */
-    lrtInputName: string;
-    /** Name of the per-step decayShrink scalar tensor_input, or null when lr is
-     *  static (decayShrink baked into the kernel) or no params are decayed. */
-    decayShrinkInputName: string | null;
-    /** Hyperparameters as captured (so the runtime can compute lrt and decayShrink). */
-    config: AdamResolvedConfig;
-}
-/**
- * Append Adam update ops to `graph`. Must be called inside an active trace
- * context (or after a trace, since traceInto re-enters the graph).
- *
- * @param graph the graph (already containing forward + backward)
- * @param paramGrads param name -> gradient tensor (output of `appendGrad`)
- * @param paramTensors param name -> the param's leaf Tensor (the param_input).
- *                     Needed because the param_input lives in the graph but we
- *                     don't have a direct map by name in `Graph` — caller passes it.
- * @param config Adam hyperparameters. Set `weightDecay > 0` for AdamW; an
- *               optional `decayFilter` selects which params receive decay.
- */
-export declare function appendAdam(graph: Graph, paramGrads: Record<string, Tensor>, paramTensors: Record<string, Tensor>, config: AdamConfig, 
-/** Per-param decay flags from `materializeParams`. When supplied, overrides
- *  `config.decayFilter` for any name in the map; falls back to `decayFilter`
- *  for names not present (e.g., for low-level callers using `compile()`
- *  directly without a Module). */
-decayFlags?: Record<string, boolean>): AdamResult;
-//# sourceMappingURL=adam.d.ts.map

package/dist/adam.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"adam.d.ts","sourceRoot":"","sources":["../src/adam.ts"],"names":[],"mappings":"AA4BA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AACrC,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,SAAS,CAAA;AACpC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,cAAc,CAAA;AAIjD,MAAM,WAAW,UAAU;IACzB;;;0DAGsD;IACtD,EAAE,EAAE,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,CAAC,CAAA;IACvC,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,EAAE,CAAC,EAAE,MAAM,CAAA;IACX,GAAG,CAAC,EAAE,MAAM,CAAA;IACZ;;6DAEyD;IACzD,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB;;;6EAGyE;IACzE,WAAW,CAAC,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK,OAAO,CAAA;CAC7C;AAED,+EAA+E;AAC/E,MAAM,WAAW,kBAAkB;IACjC,EAAE,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,MAAM,CAAA;IAC5B,EAAE,EAAE,MAAM,CAAA;IACV,EAAE,EAAE,MAAM,CAAA;IACV,GAAG,EAAE,MAAM,CAAA;IACX,WAAW,EAAE,MAAM,CAAA;IACnB,WAAW,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAA;IACtC;kEAC8D;IAC9D,aAAa,EAAE,OAAO,CAAA;CACvB;AAED,MAAM,WAAW,UAAU;IACzB,kEAAkE;IAClE,UAAU,EAAE,aAAa,EAAE,CAAA;IAC3B;iFAC6E;IAC7E,YAAY,EAAE,MAAM,CAAA;IACpB;+EAC2E;IAC3E,oBAAoB,EAAE,MAAM,GAAG,IAAI,CAAA;IACnC,oFAAoF;IACpF,MAAM,EAAE,kBAAkB,CAAA;CAC3B;AAED;;;;;;;;;;;GAWG;AACH,wBAAgB,UAAU,CACxB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAClC,YAAY,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EACpC,MAAM,EAAE,UAAU;AAClB;;;kCAGkC;AAClC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GACnC,UAAU,CAyEZ"}

package/dist/buffers.d.ts

@@ -1,57 +0,0 @@
-import type { Graph, Tensor, Dtype, Shape } from './ir.js';
-export interface BufferSpec {
-    /** Matches tensor.id. */
-    id: number;
-    byteSize: number;
-    dtype: Dtype;
-    shape: Shape;
-    kind: 'param' | 'param_grad' | 'tensor_input' | 'state' | 'intermediate' | 'output';
-    /** External name for param/param_grad/tensor_input/state bindings. null otherwise. */
-    name: string | null;
-    /** For state buffers: the value to fill on initial allocation. 0 by default. */
-    initValue?: number;
-}
-/**
- * After step(), copy `source`'s buffer into `dest`'s buffer.
- * Used to write back updated optimizer state and updated parameters into
- * their persistent home buffers.
- */
-export interface Writeback {
-    source: number;
-    dest: number;
-    bytes: number;
-}
-export interface BufferPlan {
-    buffers: BufferSpec[];
-    /** Tensor id -> buffer id (currently 1:1 but kept opaque for future pooling). */
-    tensorToBuffer: Map<number, number>;
-    /** Easy lookup tables for the runtime. */
-    paramsByName: Map<string, number>;
-    inputsByName: Map<string, number>;
-    paramGradsByName: Map<string, number>;
-    statesByName: Map<string, number>;
-    capturesByName: Map<string, number>;
-    outputBufferIds: number[];
-    /** End-of-step writebacks (Adam updates for params, m, v, etc.) */
-    writebacks: Writeback[];
-}
-/**
- * Caller-supplied writeback declarations: "after each step, copy this Tensor's
- * buffer into the persistent home of this param/state."
- */
-export interface WritebackDecl {
-    /** The Tensor (output of some op) holding the new value to write back. */
-    source: Tensor;
-    /** Either a param name (writes to that param's home buffer) or a state name. */
-    destName: string;
-    destKind: 'param' | 'state';
-}
-/**
- * Build a BufferPlan from a graph + the param-grad map produced by appendGrad.
- * @param graph the full graph (forward + backward + any optimizer ops)
- * @param paramGrads map from param name -> the Tensor that holds its gradient
- * @param writebackDecls list of end-of-step writebacks (e.g. from appendAdam).
- *                       Empty when there's no optimizer in the graph.
- */
-export declare function planBuffers(graph: Graph, paramGrads: Record<string, Tensor>, writebackDecls?: WritebackDecl[]): BufferPlan;
-//# sourceMappingURL=buffers.d.ts.map

package/dist/buffers.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"buffers.d.ts","sourceRoot":"","sources":["../src/buffers.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAU,MAAM,SAAS,CAAA;AAElE,MAAM,WAAW,UAAU;IACzB,yBAAyB;IACzB,EAAE,EAAE,MAAM,CAAA;IACV,QAAQ,EAAE,MAAM,CAAA;IAChB,KAAK,EAAE,KAAK,CAAA;IACZ,KAAK,EAAE,KAAK,CAAA;IACZ,IAAI,EAAE,OAAO,GAAG,YAAY,GAAG,cAAc,GAAG,OAAO,GAAG,cAAc,GAAG,QAAQ,CAAA;IACnF,sFAAsF;IACtF,IAAI,EAAE,MAAM,GAAG,IAAI,CAAA;IACnB,gFAAgF;IAChF,SAAS,CAAC,EAAE,MAAM,CAAA;CACnB;AAED;;;;GAIG;AACH,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,CAAA;IACd,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,MAAM,CAAA;CACd;AAED,MAAM,WAAW,UAAU;IACzB,OAAO,EAAE,UAAU,EAAE,CAAA;IACrB,iFAAiF;IACjF,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACnC,0CAA0C;IAC1C,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACjC,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACjC,gBAAgB,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACrC,YAAY,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACjC,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IACnC,eAAe,EAAE,MAAM,EAAE,CAAA;IACzB,mEAAmE;IACnE,UAAU,EAAE,SAAS,EAAE,CAAA;CACxB;AAUD;;;GAGG;AACH,MAAM,WAAW,aAAa;IAC5B,0EAA0E;IAC1E,MAAM,EAAE,MAAM,CAAA;IACd,gFAAgF;IAChF,QAAQ,EAAE,MAAM,CAAA;IAChB,QAAQ,EAAE,OAAO,GAAG,OAAO,CAAA;CAC5B;AAED;;;;;;GAMG;AACH,wBAAgB,WAAW,CACzB,KAAK,EAAE,KAAK,EACZ,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,EAClC,cAAc,GAAE,aAAa,EAAO,GACnC,UAAU,CAmGZ"}

package/dist/capture.d.ts

@@ -1,3 +0,0 @@
-import type { Tensor } from './ir.js';
-export declare function capture<T extends Tensor>(name: string, t: T): T;
-//# sourceMappingURL=capture.d.ts.map

package/dist/capture.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"capture.d.ts","sourceRoot":"","sources":["../src/capture.ts"],"names":[],"mappings":"AAqBA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,SAAS,CAAA;AAGrC,wBAAgB,OAAO,CAAC,CAAC,SAAS,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,GAAG,CAAC,CAW/D"}

package/dist/codegen.d.ts

@@ -1,23 +0,0 @@
-import type { Graph, OpNode } from './ir.js';
-import type { BufferPlan } from './buffers.js';
-export interface KernelSpec {
-    /** Index into graph.ops. */
-    opIndex: number;
-    /** Op kind (for debugging / pipeline cache key). */
-    opKind: OpNode['kind'];
-    /** Generated WGSL source. Empty string for "logical" ops with no kernel. */
-    wgsl: string;
-    /**
-     * Buffer ids in binding-index order. The runtime creates a bind group with
-     * these in @binding(0..N) on @group(0). Inputs come first (read), output last
-     * (read_write).
-     */
-    bindings: number[];
-    /** Number of threads to dispatch (1-D). 0 means "skip" (e.g. reshape no-op). */
-    threads: number;
-    /** Workgroup size; usually WG_SIZE. */
-    workgroupSize: number;
-}
-/** Generate a KernelSpec per compute op in graph.ops (in dispatch order). */
-export declare function emitKernels(graph: Graph, plan: BufferPlan): KernelSpec[];
-//# sourceMappingURL=codegen.d.ts.map

package/dist/codegen.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"codegen.d.ts","sourceRoot":"","sources":["../src/codegen.ts"],"names":[],"mappings":"AAYA,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,EAAiB,MAAM,SAAS,CAAA;AAC3D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAA;AAO9C,MAAM,WAAW,UAAU;IACzB,4BAA4B;IAC5B,OAAO,EAAE,MAAM,CAAA;IACf,oDAAoD;IACpD,MAAM,EAAE,MAAM,CAAC,MAAM,CAAC,CAAA;IACtB,4EAA4E;IAC5E,IAAI,EAAE,MAAM,CAAA;IACZ;;;;OAIG;IACH,QAAQ,EAAE,MAAM,EAAE,CAAA;IAClB,gFAAgF;IAChF,OAAO,EAAE,MAAM,CAAA;IACf,uCAAuC;IACvC,aAAa,EAAE,MAAM,CAAA;CACtB;AAMD,6EAA6E;AAC7E,wBAAgB,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,GAAG,UAAU,EAAE,CAQxE"}

package/dist/compile.d.ts

@@ -1,130 +0,0 @@
-import type { Tensor, Shape, Dtype } from './ir.js';
-import { type GradResult } from './grad.js';
-import { type AdamConfig } from './adam.js';
-import { type BufferPlan } from './buffers.js';
-import { type KernelSpec } from './codegen.js';
-import { type CompiledRuntime, type CompiledForward, type RuntimeOpts } from './runtime.js';
-import { Module } from './module.js';
-/** Declares one input tensor of the model's forward function. The name is the
- *  key in the `inputs:` Record at compile time and the key on the `step()`/
- *  `run()` data object at runtime. */
-export interface InputDecl {
-    shape: Shape;
-    dtype?: Dtype;
-}
-/** Inputs declaration: a Record from input name to its shape/dtype. The name
- *  doubles as the key the forward fn destructures and the key the runtime
- *  expects in `step({...})` / `run({...})`. */
-export type InputDecls = Record<string, InputDecl>;
-/** Maps an `InputDecls` Record to its forward-time tensor counterpart —
- *  same keys, each value is a Tensor. Used to type the forward function's
- *  `inputs` argument from the declared shape Record. */
-export type InputsTensors<I extends InputDecls> = {
-    [K in keyof I]: Tensor;
-};
-/** Forward function shape: takes the materialized model and a Record of
- *  named input tensors (matching the declared `inputs:` keys), returns the
- *  output tensor (loss for compileModule; logits/etc. for compileForward).
- *  The second generic flows from the inputs declaration so destructuring
- *  the input record stays typed. */
-export type ForwardFn<M extends Module, I extends InputDecls = InputDecls> = (m: M, inputs: InputsTensors<I>) => Tensor;
-export interface CompiledIR {
-    graph: GradResult['graph'];
-    paramGrads: GradResult['paramGrads'];
-    loss: Tensor;
-    plan: BufferPlan;
-    kernels: KernelSpec[];
-}
-/** Trace + autograd + buffer-plan + codegen, without touching WebGPU. */
-export declare function compileToIR(traceFn: () => Tensor): CompiledIR;
-/** Full compile pipeline. Browser-only because it creates a GPUDevice. */
-export declare function compile(traceFn: () => Tensor, opts?: RuntimeOpts): Promise<CompiledRuntime & {
-    ir: CompiledIR;
-}>;
-export interface CompileModuleOptions<I extends InputDecls = InputDecls> extends RuntimeOpts {
-    /** Per-step data inputs to the forward function, keyed by name. The forward
-     *  fn destructures these out of its second argument; runtime calls to
-     *  `step()` / `run()` pass typed arrays under the same keys. */
-    inputs?: I;
-    /** Adam hyperparameters. If omitted, no optimizer is appended (forward-only). */
-    adam?: AdamConfig;
-}
-export interface CompileForwardOptions<I extends InputDecls = InputDecls> extends RuntimeOpts {
-    /** Per-step data inputs to the forward function, keyed by name. */
-    inputs?: I;
-}
-/** Forward-only compile options as taken by the `compileForward` *method* on
- *  a training runtime — no `device` (inherited) and no `sharedParams`
- *  (auto-supplied from the train graph's params). */
-export interface CompileForwardMethodOptions<I extends InputDecls = InputDecls> {
-    inputs?: I;
-}
-/** Returned by `compileModule`. Adds training-graph extras (auto-init, reset,
- *  sibling-graph compile) on top of the base runtime. */
-export interface CompiledModule<M extends Module> extends CompiledRuntime {
-    ir: CompiledIR;
-    /** Number of dispatchable kernels (excludes leaf no-ops). */
-    kernelCount: number;
-    /** Re-initialize all params from their declared init specs and zero the
-     *  optimizer state. Use to start training over without recompiling. */
-    reset(): void;
-    /** Compile a sibling forward-only graph (e.g., a B=1 inference graph or a
-     *  B=N held-out eval graph) that shares this runtime's device and param
-     *  buffers. Pass the forward fn (typically distinct from your loss fn —
-     *  it returns logits, not a scalar) and any shape changes via `inputs`.
-     *  Auto-initialization is a no-op since params are shared. */
-    compileForward<I extends InputDecls>(forward: ForwardFn<M, I>, opts?: CompileForwardMethodOptions<I>): Promise<CompiledForwardModule>;
-}
-/** Returned by `compileForward` (and by the `compileForward` method). */
-export interface CompiledForwardModule extends CompiledForward {
-    ir: CompiledIR;
-    /** Number of dispatchable kernels (excludes leaf no-ops). */
-    kernelCount: number;
-}
-/**
- * Compile a Module-based model. Pass a *factory* `() => new Model()`, not the
- * model instance itself: compilation mutates the tree (every `ParamSentinel`
- * field becomes a real `Tensor`), so the instance is consumed and shouldn't be
- * referenced afterwards. Re-call the factory if you need a fresh tree.
- *
- * The forward function takes the materialized model and a Record of named
- * input tensors, returns the loss tensor. Inputs are matched by name with the
- * `inputs:` declaration:
- *
- *   inputs: {
- *     tokens:  { shape: [B, T], dtype: 'i32' },
- *     targets: { shape: [B, T], dtype: 'i32' },
- *   }
- *   forward: (m, { tokens, targets }) => …
- *
- * Walks the module tree to materialize params with auto-derived names, then
- * runs trace → grad → adam → buffer plan → codegen → runtime. Initial
- * parameter values are uploaded automatically before this function returns;
- * call `reset()` later to re-randomize.
- *
- * If `opts.adam` is set, the runtime's `step()` automatically tracks an
- * internal step count and injects the bias-corrected `lrt` scalar each call;
- * users don't need to provide it themselves.
- */
-export declare function compileModule<M extends Module, I extends InputDecls = InputDecls>(modelFactory: () => M, forward: ForwardFn<M, I>, opts?: CompileModuleOptions<I>): Promise<CompiledModule<M>>;
-/**
- * Compile a Module-based model in forward-only mode (no autograd, no Adam).
- * The forward function returns the output tensor (e.g., logits) instead of a
- * scalar loss; runtime exposes `run(inputs)` returning the full output as a
- * `Float32Array`.
- *
- * **Prefer the `compileForward` method on a training runtime** when both
- * graphs use the same Module class — it auto-supplies `device` and
- * `sharedParams`. This standalone form is for forward-only models with no
- * training graph at all, or for sharing params across a different model.
- *
- * **Sharing params with a training compile.** Pass `opts.sharedParams =
- * trainCompiled.params` to bind this graph's param buffers to an existing
- * training runtime's GPU buffers — every train step is then immediately
- * visible to `run()` calls here, no copies.
- *
- * Initial param values are uploaded automatically for params *not* covered
- * by `sharedParams` (those are owned by the sibling compile).
- */
-export declare function compileForward<M extends Module, I extends InputDecls = InputDecls>(modelFactory: () => M, forward: ForwardFn<M, I>, opts?: CompileForwardOptions<I>): Promise<CompiledForwardModule>;
-//# sourceMappingURL=compile.d.ts.map

package/dist/compile.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"compile.d.ts","sourceRoot":"","sources":["../src/compile.ts"],"names":[],"mappings":"AAUA,OAAO,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,SAAS,CAAA;AAEnD,OAAO,EAAc,KAAK,UAAU,EAAE,MAAM,WAAW,CAAA;AACvD,OAAO,EAAc,KAAK,UAAU,EAAE,MAAM,WAAW,CAAA;AACvD,OAAO,EAAe,KAAK,UAAU,EAAE,MAAM,cAAc,CAAA;AAC3D,OAAO,EAAe,KAAK,UAAU,EAAE,MAAM,cAAc,CAAA;AAC3D,OAAO,EAAuC,KAAK,eAAe,EAAE,KAAK,eAAe,EAAE,KAAK,WAAW,EAAE,MAAM,cAAc,CAAA;AAChI,OAAO,EAAE,MAAM,EAAqB,MAAM,aAAa,CAAA;AAEvD;;sCAEsC;AACtC,MAAM,WAAW,SAAS;IACxB,KAAK,EAAE,KAAK,CAAA;IACZ,KAAK,CAAC,EAAE,KAAK,CAAA;CACd;AAED;;+CAE+C;AAC/C,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,CAAA;AAElD;;wDAEwD;AACxD,MAAM,MAAM,aAAa,CAAC,CAAC,SAAS,UAAU,IAAI;KAAG,CAAC,IAAI,MAAM,CAAC,GAAG,MAAM;CAAE,CAAA;AAE5E;;;;oCAIoC;AACpC,MAAM,MAAM,SAAS,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,IACvE,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC,KAAK,MAAM,CAAA;AAE5C,MAAM,WAAW,UAAU;IACzB,KAAK,EAAE,UAAU,CAAC,OAAO,CAAC,CAAA;IAC1B,UAAU,EAAE,UAAU,CAAC,YAAY,CAAC,CAAA;IACpC,IAAI,EAAE,MAAM,CAAA;IACZ,IAAI,EAAE,UAAU,CAAA;IAChB,OAAO,EAAE,UAAU,EAAE,CAAA;CACtB;AAED,yEAAyE;AACzE,wBAAgB,WAAW,CAAC,OAAO,EAAE,MAAM,MAAM,GAAG,UAAU,CAM7D;AAED,0EAA0E;AAC1E,wBAAsB,OAAO,CAAC,OAAO,EAAE,MAAM,MAAM,EAAE,IAAI,GAAE,WAAgB,GAAG,OAAO,CAAC,eAAe,GAAG;IAAE,EAAE,EAAE,UAAU,CAAA;CAAE,CAAC,CAK1H;AAMD,MAAM,WAAW,oBAAoB,CAAC,CAAC,SAAS,UAAU,GAAG,UAAU,CAAE,SAAQ,WAAW;IAC1F;;oEAEgE;IAChE,MAAM,CAAC,EAAE,CAAC,CAAA;IACV,iFAAiF;IACjF,IAAI,CAAC,EAAE,UAAU,CAAA;CAClB;AAED,MAAM,WAAW,qBAAqB,CAAC,CAAC,SAAS,UAAU,GAAG,UAAU,CAAE,SAAQ,WAAW;IAC3F,mEAAmE;IACnE,MAAM,CAAC,EAAE,CAAC,CAAA;CACX;AAED;;qDAEqD;AACrD,MAAM,WAAW,2BAA2B,CAAC,CAAC,SAAS,UAAU,GAAG,UAAU;IAC5E,MAAM,CAAC,EAAE,CAAC,CAAA;CACX;AAED;yDACyD;AACzD,MAAM,WAAW,cAAc,CAAC,CAAC,SAAS,MAAM,CAAE,SAAQ,eAAe;IACvE,EAAE,EAAE,UAAU,CAAA;IACd,6DAA6D;IAC7D,WAAW,EAAE,MAAM,CAAA;IACnB;2EACuE;IACvE,KAAK,IAAI,IAAI,CAAA;IACb;;;;kEAI8D;IAC9D,cAAc,CAAC,CAAC,SAAS,UAAU,EACjC,OAAO,EAAE,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EACxB,IAAI,CAAC,EAAE,2BAA2B,CAAC,CAAC,CAAC,GACpC,OAAO,CAAC,qBAAqB,CAAC,CAAA;CAClC;AAED,yEAAyE;AACzE,MAAM,WAAW,qBAAsB,SAAQ,eAAe;IAC5D,EAAE,EAAE,UAAU,CAAA;IACd,6DAA6D;IAC7D,WAAW,EAAE,MAAM,CAAA;CACpB;AAED;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,wBAAsB,aAAa,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EACrF,YAAY,EAAE,MAAM,CAAC,EACrB,OAAO,EAAE,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EACxB,IAAI,GAAE,oBAAoB,CAAC,CAAC,CAAM,GACjC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAyC5B;AAMD;;;;;;;;;;;;;;;;;;GAkBG;AACH,wBAAsB,cAAc,CAAC,CAAC,SAAS,MAAM,EAAE,CAAC,SAAS,UAAU,GAAG,UAAU,EACtF,YAAY,EAAE,MAAM,CAAC,EACrB,OAAO,EAAE,SAAS,CAAC,CAAC,EAAE,CAAC,CAAC,EACxB,IAAI,GAAE,qBAAqB,CAAC,CAAC,CAAM,GAClC,OAAO,CAAC,qBAAqB,CAAC,CAYhC"}

package/dist/grad.d.ts

@@ -1,8 +0,0 @@
-import type { Graph, Tensor } from './ir.js';
-export interface GradResult {
-    readonly graph: Graph;
-    readonly paramGrads: Record<string, Tensor>;
-    readonly loss: Tensor;
-}
-export declare function appendGrad(graph: Graph): GradResult;
-//# sourceMappingURL=grad.d.ts.map

package/dist/grad.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"grad.d.ts","sourceRoot":"","sources":["../src/grad.ts"],"names":[],"mappings":"AAiBA,OAAO,KAAK,EAAE,KAAK,EAAU,MAAM,EAAS,MAAM,SAAS,CAAA;AAe3D,MAAM,WAAW,UAAU;IAEzB,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAA;IAErB,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAE3C,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAA;CACtB;AASD,wBAAgB,UAAU,CAAC,KAAK,EAAE,KAAK,GAAG,UAAU,CAmDnD"}

package/dist/index.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAKA,YAAY,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAA;AAC5E,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AACvC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AAClF,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAA;AACtC,OAAO,EAEL,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAElB,IAAI,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAE3B,IAAI,EAAE,OAAO,EAAE,KAAK,EAEpB,QAAQ,EAAE,OAAO,EAAE,MAAM,EAEzB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAE5B,MAAM,EAAE,aAAa,EAErB,MAAM,EAAE,MAAM,EAAE,SAAS,EAEzB,iBAAiB,EAAE,cAAc,EAAE,WAAW,EAE9C,cAAc,GACf,MAAM,UAAU,CAAA;AAMjB,OAAO,EAAE,UAAU,EAAE,KAAK,UAAU,EAAE,MAAM,WAAW,CAAA;AACvD,OAAO,EAAE,UAAU,EAAE,KAAK,UAAU,EAAE,KAAK,UAAU,EAAE,MAAM,WAAW,CAAA;AACxE,OAAO,EAAE,WAAW,EAAE,KAAK,UAAU,EAAE,KAAK,UAAU,EAAE,KAAK,SAAS,EAAE,KAAK,aAAa,EAAE,MAAM,cAAc,CAAA;AAChH,OAAO,EAAE,WAAW,EAAE,KAAK,UAAU,EAAE,MAAM,cAAc,CAAA;AAC3D,OAAO,EAAE,aAAa,EAAE,oBAAoB,EAAE,QAAQ,EAAE,KAAK,eAAe,EAAE,KAAK,eAAe,EAAE,KAAK,WAAW,EAAE,KAAK,UAAU,EAAE,KAAK,UAAU,EAAE,KAAK,SAAS,EAAE,MAAM,cAAc,CAAA;AAC5L,OAAO,EACL,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,cAAc,EACnD,KAAK,UAAU,EAAE,KAAK,oBAAoB,EAAE,KAAK,qBAAqB,EAAE,KAAK,2BAA2B,EACxG,KAAK,cAAc,EAAE,KAAK,qBAAqB,EAC/C,KAAK,SAAS,EAAE,KAAK,UAAU,EAAE,KAAK,aAAa,EAAE,KAAK,SAAS,GACpE,MAAM,cAAc,CAAA;AACrB,OAAO,EAAE,MAAM,EAAE,iBAAiB,EAAE,KAAK,QAAQ,EAAE,KAAK,YAAY,EAAE,KAAK,kBAAkB,EAAE,MAAM,aAAa,CAAA;AAClH,OAAO,KAAK,EAAE,MAAM,SAAS,CAAA"}

package/dist/ir.d.ts

@@ -1,207 +0,0 @@
-export type Dtype = 'f32' | 'i32' | 'bool';
-export type Shape = readonly number[];
-export interface Tensor {
-    readonly id: number;
-    readonly shape: Shape;
-    readonly dtype: Dtype;
-    readonly source: number | null;
-    readonly site: CallSite | null;
-}
-export interface CallSite {
-    readonly opName: string;
-    readonly stack: string;
-}
-export type OpNode = {
-    kind: 'param_input';
-    out: number;
-    name: string;
-} | {
-    kind: 'tensor_input';
-    out: number;
-    name: string;
-} | {
-    kind: 'state_input';
-    out: number;
-    name: string;
-    initValue: number;
-} | {
-    kind: 'add';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'sub';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'mul';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'div';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'mul_scalar';
-    out: number;
-    a: number;
-    scalar: number;
-} | {
-    kind: 'add_scalar';
-    out: number;
-    a: number;
-    scalar: number;
-} | {
-    kind: 'sqrt';
-    out: number;
-    a: number;
-} | {
-    kind: 'rsqrt';
-    out: number;
-    a: number;
-} | {
-    kind: 'log';
-    out: number;
-    a: number;
-} | {
-    kind: 'exp';
-    out: number;
-    a: number;
-} | {
-    kind: 'relu';
-    out: number;
-    a: number;
-} | {
-    kind: 'mean_last';
-    out: number;
-    a: number;
-} | {
-    kind: 'sum_last';
-    out: number;
-    a: number;
-} | {
-    kind: 'reshape';
-    out: number;
-    a: number;
-    newShape: Shape;
-} | {
-    kind: 'transpose';
-    out: number;
-    a: number;
-    perm: readonly number[];
-} | {
-    kind: 'matmul';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'matmul_batched';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'one_hot';
-    out: number;
-    indices: number;
-    depth: number;
-    dtype: Dtype;
-} | {
-    kind: 'arange';
-    out: number;
-    n: number;
-    dtype: Dtype;
-} | {
-    kind: 'softmax_causal_last';
-    out: number;
-    a: number;
-} | {
-    kind: 'log_softmax_last';
-    out: number;
-    a: number;
-} | {
-    kind: 'where_causal';
-    out: number;
-    a: number;
-    fillValue: number;
-} | {
-    kind: 'less';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'greater';
-    out: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'where';
-    out: number;
-    cond: number;
-    a: number;
-    b: number;
-} | {
-    kind: 'adam_update_m';
-    out: number;
-    m: number;
-    g: number;
-    b1: number;
-} | {
-    kind: 'adam_update_v';
-    out: number;
-    v: number;
-    g: number;
-    b2: number;
-} | {
-    kind: 'adam_update_p';
-    out: number;
-    p: number;
-    mNew: number;
-    vNew: number;
-    lrt: number;
-    eps: number;
-    decayShrink: number;
-    decayShrinkTensor: number | null;
-} | {
-    kind: 'slice_last_range';
-    out: number;
-    a: number;
-    start: number;
-    end: number;
-} | {
-    kind: 'broadcast_to';
-    out: number;
-    a: number;
-    targetShape: Shape;
-} | {
-    kind: 'sum_to_shape';
-    out: number;
-    a: number;
-    targetShape: Shape;
-} | {
-    kind: 'const_scalar';
-    out: number;
-    value: number;
-    dtype: Dtype;
-} | {
-    kind: 'relu_grad';
-    out: number;
-    x: number;
-    dy: number;
-};
-export interface Graph {
-    readonly ops: OpNode[];
-    readonly tensors: Tensor[];
-    readonly outputs: number[];
-    readonly captures: Map<string, number>;
-}
-export declare function makeGraph(): Graph;
-export declare function addTensor(g: Graph, shape: Shape, dtype: Dtype, source: number | null, site: CallSite | null): Tensor;
-export declare function addOp<K extends OpNode['kind']>(g: Graph, kind: K, shape: Shape, dtype: Dtype, site: CallSite | null, fields: Omit<Extract<OpNode, {
-    kind: K;
-}>, 'kind' | 'out'>): Tensor;
-export declare function captureSite(opName: string): CallSite;
-export declare function formatSite(site: CallSite): string;
-//# sourceMappingURL=ir.d.ts.map

package/dist/ir.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"ir.d.ts","sourceRoot":"","sources":["../src/ir.ts"],"names":[],"mappings":"AAcA,MAAM,MAAM,KAAK,GAAG,KAAK,GAAG,KAAK,GAAG,MAAM,CAAA;AAC1C,MAAM,MAAM,KAAK,GAAG,SAAS,MAAM,EAAE,CAAA;AAIrC,MAAM,WAAW,MAAM;IACrB,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAA;IACnB,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAA;IACrB,QAAQ,CAAC,KAAK,EAAE,KAAK,CAAA;IAErB,QAAQ,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAA;IAG9B,QAAQ,CAAC,IAAI,EAAE,QAAQ,GAAG,IAAI,CAAA;CAC/B;AAED,MAAM,WAAW,QAAQ;IACvB,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAA;IAEvB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAA;CACvB;AAQD,MAAM,MAAM,MAAM,GAGd;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAElD;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAInD;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GAGrE;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAClD;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAClD;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAClD;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAClD;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAC9D;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GAG9D;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GACxC;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GACzC;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GACvC;IAAE,IAAI,EAAE,KAAK,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GACvC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAGxC;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAC7C;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAG5C;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,KAAK,CAAA;CAAE,GAC5D;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,SAAS,MAAM,EAAE,CAAA;CAAE,GAMtE;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAErD;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAG7D;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GAC9E;IAAE,IAAI,EAAE,QAAQ,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GAGxD;IAAE,IAAI,EAAE,qBAAqB,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GACvD;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAIpD;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAA;CAAE,GAKnE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GACnD;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAGtD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,GAMlE;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAA;CAAE,GACxE;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAA;CAAE,GASxE;IACE,IAAI,EAAE,eAAe,CAAA;IACrB,GAAG,EAAE,MAAM,CAAA;IACX,CAAC,EAAE,MAAM,CAAA;IACT,IAAI,EAAE,MAAM,CAAA;IACZ,IAAI,EAAE,MAAM,CAAA;IACZ,GAAG,EAAE,MAAM,CAAA;IACX,GAAG,EAAE,MAAM,CAAA;IACX,WAAW,EAAE,MAAM,CAAA;IACnB,iBAAiB,EAAE,MAAM,GAAG,IAAI,CAAA;CACjC,GAMD;IAAE,IAAI,EAAE,kBAAkB,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAE,GAGhF;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,KAAK,CAAA;CAAE,GAGpE;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,KAAK,CAAA;CAAE,GAEpE;IAAE,IAAI,EAAE,cAAc,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,KAAK,CAAA;CAAE,GAElE;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAA;CAAE,CAAA;AAI7D,MAAM,WAAW,KAAK;IACpB,QAAQ,CAAC,GAAG,EAAE,MAAM,EAAE,CAAA;IACtB,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,CAAA;IAG1B,QAAQ,CAAC,OAAO,EAAE,MAAM,EAAE,CAAA;IAI1B,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;CACvC;AAED,wBAAgB,SAAS,IAAI,KAAK,CAEjC;AAGD,wBAAgB,SAAS,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,GAAG,IAAI,EAAE,IAAI,EAAE,QAAQ,GAAG,IAAI,GAAG,MAAM,CAKpH;AAMD,wBAAgB,KAAK,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,CAAC,EAC5C,CAAC,EAAE,KAAK,EACR,IAAI,EAAE,CAAC,EACP,KAAK,EAAE,KAAK,EACZ,KAAK,EAAE,KAAK,EACZ,IAAI,EAAE,QAAQ,GAAG,IAAI,EACrB,MAAM,EAAE,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE;IAAE,IAAI,EAAE,CAAC,CAAA;CAAE,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,GACzD,MAAM,CAMR;AAID,wBAAgB,WAAW,CAAC,MAAM,EAAE,MAAM,GAAG,QAAQ,CAIpD;AAID,wBAAgB,UAAU,CAAC,IAAI,EAAE,QAAQ,GAAG,MAAM,CAYjD"}

package/dist/module.d.ts

@@ -1,55 +0,0 @@
-import type { Tensor, Shape, Dtype } from './ir.js';
-/** How a parameter's initial values are produced.
- *  - `'randn'` — Gaussian, with `scale` (default 0.02). The common case for
- *    weight matrices and embeddings.
- *  - `'zeros'` — fill with 0. Common for biases and LayerNorm beta.
- *  - `'ones'`  — fill with 1. Common for LayerNorm gain.
- *  - Custom function — receives total element count and shape, returns the
- *    Float32Array. Use for fan-in scaling or any non-standard scheme.
- */
-export type InitSpec = 'randn' | 'zeros' | 'ones' | ((size: number, shape: readonly number[]) => Float32Array);
-export interface ParamOptions {
-    dtype?: Dtype;
-    /** Init kind. Default: `'randn'`. */
-    init?: InitSpec;
-    /** Std dev for `'randn'`. Default 0.02. Ignored for non-randn init. */
-    scale?: number;
-    /** Whether AdamW (when `weightDecay > 0`) should apply decoupled weight
-     *  decay to this param. Default: `true` for `'randn'` init (weight matrices,
-     *  embeddings), `false` for `'zeros'` / `'ones'` (biases, LN gains). Override
-     *  to force or skip. Replaces `adam.decayFilter` for the common case. */
-    decay?: boolean;
-}
-type InitFn = (size: number, shape: readonly number[]) => Float32Array;
-export declare abstract class Module {
-    /**
-     * Declare a learnable parameter at this module. Must be called from inside
-     * the constructor (typically as a field assignment). Returns a placeholder
-     * that gets replaced with a real Tensor at compile time.
-     *
-     * The parameter's name is auto-derived from its property path in the model
-     * tree (e.g. `layers.0.attn.W_q`). Init metadata travels with the param;
-     * call `compiled.uploadInitialParams()` to apply it after compile.
-     */
-    protected param(shape: Shape, opts?: ParamOptions): Tensor;
-}
-export interface MaterializedParams {
-    /** Map from auto-derived path (e.g. `layers.0.attn.W_q`) to its Tensor. */
-    tensors: Record<string, Tensor>;
-    /** Init function per param path. Used by `uploadInitialParams`. */
-    initFns: Record<string, InitFn>;
-    /** Whether this param should receive AdamW weight decay. Resolved at
-     *  `param()` time from `ParamOptions.decay` (with init-based default). */
-    decayFlags: Record<string, boolean>;
-}
-/**
- * Walk the module tree and replace every ParamSentinel with a real Tensor
- * created via `paramInput(autoName, ...)`. Must be called inside an active
- * trace context (paramInput appends to the current graph).
- *
- * Returns the param tensors keyed by path, plus init functions for use by
- * `uploadInitialParams`.
- */
-export declare function materializeParams(root: Module): MaterializedParams;
-export {};
-//# sourceMappingURL=module.d.ts.map

package/dist/module.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"module.d.ts","sourceRoot":"","sources":["../src/module.ts"],"names":[],"mappings":"AA2BA,OAAO,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,SAAS,CAAA;AAOnD;;;;;;;GAOG;AACH,MAAM,MAAM,QAAQ,GAChB,OAAO,GACP,OAAO,GACP,MAAM,GACN,CAAC,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,MAAM,EAAE,KAAK,YAAY,CAAC,CAAA;AAE9D,MAAM,WAAW,YAAY;IAC3B,KAAK,CAAC,EAAE,KAAK,CAAA;IACb,qCAAqC;IACrC,IAAI,CAAC,EAAE,QAAQ,CAAA;IACf,uEAAuE;IACvE,KAAK,CAAC,EAAE,MAAM,CAAA;IACd;;;6EAGyE;IACzE,KAAK,CAAC,EAAE,OAAO,CAAA;CAChB;AAED,KAAK,MAAM,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,SAAS,MAAM,EAAE,KAAK,YAAY,CAAA;AAuDtE,8BAAsB,MAAM;IAC1B;;;;;;;;OAQG;IACH,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,EAAE,YAAY,GAAG,MAAM;CAK3D;AAMD,MAAM,WAAW,kBAAkB;IACjC,2EAA2E;IAC3E,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC/B,mEAAmE;IACnE,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAA;IAC/B;8EAC0E;IAC1E,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CACpC;AAED;;;;;;;GAOG;AACH,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,kBAAkB,CAclE"}