@dxo/train 0.0.8 → 0.0.10

package/README.md

@@ -1,4 +1,4 @@
-# @dxo/train
+# 🏋️ @dxo/train
 
 **Developer preview — API unstable (`0.0.x`).**
 
@@ -6,13 +6,13 @@ Async training loop over `@dxo/core` / `@dxo/nn` / `@dxo/optimizer` / `@dxo/data
 
 ## Contract (0.0.7 / G5)
 
-| API | Behavior |
-|-----|----------|
-| `Trainer` | Owns model + optimizer + batch factory + epoch count |
-| `fitIter` / `run` | `AsyncGenerator<TrainEvent>`; respects `AbortSignal` |
-| `fit` | Consumes the event stream; returns a short summary |
-| Checkpoint | Emits `encodeLinearState(model.state())` documents (no FS I/O) |
-| Device | **CPU-only** in this slice (GPU spike not required for this gate) |
+| API               | Behavior                                                          |
+|-------------------|-------------------------------------------------------------------|
+| `Trainer`         | Owns model + optimizer + batch factory + epoch count              |
+| `fitIter` / `run` | `AsyncGenerator<TrainEvent>`; respects `AbortSignal`              |
+| `fit`             | Consumes the event stream; returns a short summary                |
+| Checkpoint        | Emits `encodeLinearState(model.state())` documents (no FS I/O)    |
+| Device            | **CPU-only** in this slice (GPU spike not required for this gate) |
 
 ```typescript
 import { batch, dataset } from '@dxo/data';

package/dist/index.d.ts

@@ -0,0 +1,91 @@
+import type { Tensor } from '@dxo/core';
+import type { Batch } from '@dxo/data';
+import type { Linear, LinearState } from '@dxo/nn';
+import type { Optimizer } from '@dxo/optimizer';
+import { type StateDocument } from '@dxo/serialize';
+/** Events yielded by {@link Trainer.fitIter} / {@link Trainer.run}. */
+export type TrainEvent = {
+    type: 'epoch_start';
+    epoch: number;
+    epochs: number;
+} | {
+    type: 'batch';
+    epoch: number;
+    step: number;
+    loss: number;
+} | {
+    type: 'epoch_end';
+    epoch: number;
+    meanLoss: number;
+    steps: number;
+} | {
+    type: 'checkpoint';
+    epoch: number;
+    document: StateDocument;
+    state: LinearState;
+} | {
+    type: 'aborted';
+    reason: 'signal';
+    epoch: number;
+    step: number;
+} | {
+    type: 'done';
+    epochs: number;
+    steps: number;
+    finalMeanLoss?: number;
+};
+export interface FitSummary {
+    epochs: number;
+    steps: number;
+    aborted: boolean;
+    finalMeanLoss?: number;
+    /** Last checkpoint document, if any was emitted. */
+    lastCheckpoint?: StateDocument;
+}
+export type BatchSource = Iterable<Batch> | AsyncIterable<Batch>;
+export interface TrainerOptions {
+    model: Linear;
+    optimizer: Optimizer;
+    /** Called each epoch so iterators restart (sync or async batches). */
+    batches: () => BatchSource;
+    epochs: number;
+    /**
+     * Scalar loss from prediction and target. Default: mean squared error.
+     * Must return a tensor with `numel === 1`.
+     */
+    loss?: (pred: Tensor, y: Tensor) => Tensor;
+    /** Emit a checkpoint every N epochs (and always after the last completed epoch). Default: every epoch. */
+    checkpointEvery?: number;
+}
+/** Default MSE: `mean((pred - y)^2)`. */
+export declare function mseLoss(pred: Tensor, y: Tensor): Tensor;
+/**
+ * Minimal CPU training loop (G5 / 0.0.7).
+ *
+ * GPU / multi-device is out of scope; declare CPU-only when GPU is deferred.
+ */
+export declare class Trainer {
+    readonly model: Linear;
+    readonly optimizer: Optimizer;
+    readonly epochs: number;
+    readonly checkpointEvery: number;
+    private readonly batches;
+    private readonly lossFn;
+    constructor(options: TrainerOptions);
+    /** Alias for {@link fitIter} (Living API name). */
+    run(opts?: {
+        signal?: AbortSignal;
+    }): AsyncGenerator<TrainEvent, void, undefined>;
+    /**
+     * Yield training events. Honors `signal.aborted` between batches.
+     * Checkpoint payloads are in-memory `dxo-state` documents — callers persist them.
+     */
+    fitIter(opts?: {
+        signal?: AbortSignal;
+    }): AsyncGenerator<TrainEvent, void, undefined>;
+    /** Consume {@link fitIter} and return a summary (including last checkpoint document). */
+    fit(opts?: {
+        signal?: AbortSignal;
+    }): Promise<FitSummary>;
+}
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,WAAW,CAAC;AACxC,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,WAAW,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAChD,OAAO,EAAqB,KAAK,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAEvE,uEAAuE;AACvE,MAAM,MAAM,UAAU,GAChB;IAAE,IAAI,EAAE,aAAa,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAA;CAAE,GACtD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAC5D;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,GACrE;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,aAAa,CAAC;IAAC,KAAK,EAAE,WAAW,CAAA;CAAE,GAClF;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,MAAM,EAAE,QAAQ,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,GAClE;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,aAAa,CAAC,EAAE,MAAM,CAAA;CAAE,CAAC;AAE9E,MAAM,WAAW,UAAU;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,OAAO,CAAC;IACjB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,oDAAoD;IACpD,cAAc,CAAC,EAAE,aAAa,CAAC;CAClC;AAED,MAAM,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;AAEjE,MAAM,WAAW,cAAc;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,SAAS,CAAC;IACrB,sEAAsE;IACtE,OAAO,EAAE,MAAM,WAAW,CAAC;IAC3B,MAAM,EAAE,MAAM,CAAC;IACf;;;OAGG;IACH,IAAI,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,KAAK,MAAM,CAAC;IAC3C,0GAA0G;IAC1G,eAAe,CAAC,EAAE,MAAM,CAAC;CAC5B;AAcD,yCAAyC;AACzC,wBAAgB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,MAAM,GAAG,MAAM,CAGvD;AAED;;;;GAIG;AACH,qBAAa,OAAO;IAChB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC;IAC9B,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,eAAe,EAAE,MAAM,CAAC;IACjC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAoB;IAC5C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAsC;gBAEjD,OAAO,EAAE,cAAc;IAgBnC,mDAAmD;IACnD,GAAG,CAAC,IAAI,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GAAG,cAAc,CAAC,UAAU,EAAE,IAAI,EAAE,SAAS,CAAC;IAIjF;;;OAGG;IACI,OAAO,CAAC,IAAI,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GAAG,cAAc,CAAC,UAAU,EAAE,IAAI,EAAE,SAAS,CAAC;IA+D5F,yFAAyF;IACnF,GAAG,CAAC,IAAI,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,UAAU,CAAC;CAoClE"}

package/dist/index.js

@@ -0,0 +1,142 @@
+import { encodeLinearState } from '@dxo/serialize';
+function isAsyncIterable(source) {
+    return typeof source[Symbol.asyncIterator] === 'function';
+}
+async function* iterateBatches(source) {
+    if (isAsyncIterable(source)) {
+        for await (const b of source)
+            yield b;
+        return;
+    }
+    for (const b of source)
+        yield b;
+}
+/** Default MSE: `mean((pred - y)^2)`. */
+export function mseLoss(pred, y) {
+    const diff = pred.sub(y);
+    return diff.mul(diff).mean();
+}
+/**
+ * Minimal CPU training loop (G5 / 0.0.7).
+ *
+ * GPU / multi-device is out of scope; declare CPU-only when GPU is deferred.
+ */
+export class Trainer {
+    model;
+    optimizer;
+    epochs;
+    checkpointEvery;
+    batches;
+    lossFn;
+    constructor(options) {
+        if (!(options.epochs > 0) || !Number.isInteger(options.epochs)) {
+            throw new Error('epochs must be a positive integer');
+        }
+        const every = options.checkpointEvery ?? 1;
+        if (!(every > 0) || !Number.isInteger(every)) {
+            throw new Error('checkpointEvery must be a positive integer');
+        }
+        this.model = options.model;
+        this.optimizer = options.optimizer;
+        this.epochs = options.epochs;
+        this.checkpointEvery = every;
+        this.batches = options.batches;
+        this.lossFn = options.loss ?? mseLoss;
+    }
+    /** Alias for {@link fitIter} (Living API name). */
+    run(opts) {
+        return this.fitIter(opts);
+    }
+    /**
+     * Yield training events. Honors `signal.aborted` between batches.
+     * Checkpoint payloads are in-memory `dxo-state` documents — callers persist them.
+     */
+    async *fitIter(opts) {
+        const signal = opts?.signal;
+        let globalStep = 0;
+        let lastMean;
+        for (let epoch = 1; epoch <= this.epochs; epoch++) {
+            if (signal?.aborted) {
+                yield { type: 'aborted', reason: 'signal', epoch, step: globalStep };
+                return;
+            }
+            yield { type: 'epoch_start', epoch, epochs: this.epochs };
+            let epochLossSum = 0;
+            let epochSteps = 0;
+            for await (const batch of iterateBatches(this.batches())) {
+                if (signal?.aborted) {
+                    yield { type: 'aborted', reason: 'signal', epoch, step: globalStep };
+                    return;
+                }
+                if (!batch.y) {
+                    throw new Error('Trainer requires batch.y targets');
+                }
+                this.model.zeroGrad();
+                const pred = this.model.forward(batch.x);
+                const loss = this.lossFn(pred, batch.y);
+                const value = await loss.item();
+                if (!Number.isFinite(value)) {
+                    throw new Error(`non-finite loss at epoch ${epoch} step ${globalStep + 1}: ${value}`);
+                }
+                loss.backward();
+                this.model.loadParameters(await this.optimizer.step(this.model.parameters()));
+                globalStep += 1;
+                epochSteps += 1;
+                epochLossSum += value;
+                yield { type: 'batch', epoch, step: globalStep, loss: value };
+            }
+            if (epochSteps === 0) {
+                throw new Error(`epoch ${epoch} produced zero batches`);
+            }
+            lastMean = epochLossSum / epochSteps;
+            yield { type: 'epoch_end', epoch, meanLoss: lastMean, steps: epochSteps };
+            const shouldCheckpoint = epoch % this.checkpointEvery === 0 || epoch === this.epochs;
+            if (shouldCheckpoint) {
+                const state = await this.model.state();
+                yield {
+                    type: 'checkpoint',
+                    epoch,
+                    state,
+                    document: encodeLinearState(state),
+                };
+            }
+        }
+        yield { type: 'done', epochs: this.epochs, steps: globalStep, finalMeanLoss: lastMean };
+    }
+    /** Consume {@link fitIter} and return a summary (including last checkpoint document). */
+    async fit(opts) {
+        let epochs = 0;
+        let steps = 0;
+        let aborted = false;
+        let finalMeanLoss;
+        let lastCheckpoint;
+        for await (const event of this.fitIter(opts)) {
+            switch (event.type) {
+                case 'batch':
+                    steps = event.step;
+                    break;
+                case 'epoch_end':
+                    epochs = event.epoch;
+                    finalMeanLoss = event.meanLoss;
+                    break;
+                case 'checkpoint':
+                    lastCheckpoint = event.document;
+                    break;
+                case 'aborted':
+                    aborted = true;
+                    epochs = event.epoch;
+                    steps = event.step;
+                    break;
+                case 'done':
+                    epochs = event.epochs;
+                    steps = event.steps;
+                    finalMeanLoss = event.finalMeanLoss;
+                    break;
+                default:
+                    break;
+            }
+        }
+        return { epochs, steps, aborted, finalMeanLoss, lastCheckpoint };
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,iBAAiB,EAAsB,MAAM,gBAAgB,CAAC;AAqCvE,SAAS,eAAe,CAAC,MAAmB;IACxC,OAAO,OAAQ,MAA+B,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,UAAU,CAAC;AACxF,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,cAAc,CAAC,MAAmB;IAC9C,IAAI,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC;QAC1B,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM;YAAE,MAAM,CAAC,CAAC;QACtC,OAAO;IACX,CAAC;IACD,KAAK,MAAM,CAAC,IAAI,MAAM;QAAE,MAAM,CAAC,CAAC;AACpC,CAAC;AAED,yCAAyC;AACzC,MAAM,UAAU,OAAO,CAAC,IAAY,EAAE,CAAS;IAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACzB,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC;AACjC,CAAC;AAED;;;;GAIG;AACH,MAAM,OAAO,OAAO;IACP,KAAK,CAAS;IACd,SAAS,CAAY;IACrB,MAAM,CAAS;IACf,eAAe,CAAS;IAChB,OAAO,CAAoB;IAC3B,MAAM,CAAsC;IAE7D,YAAY,OAAuB;QAC/B,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7D,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;QACzD,CAAC;QACD,MAAM,KAAK,GAAG,OAAO,CAAC,eAAe,IAAI,CAAC,CAAC;QAC3C,IAAI,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE,CAAC;YAC3C,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAClE,CAAC;QACD,IAAI,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC7B,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAC/B,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC;IAC1C,CAAC;IAED,mDAAmD;IACnD,GAAG,CAAC,IAA+B;QAC/B,OAAO,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IAC9B,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,CAAC,OAAO,CAAC,IAA+B;QAC1C,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,CAAC;QAC5B,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,QAA4B,CAAC;QAEjC,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,IAAI,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,CAAC;YAChD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBAClB,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;gBACrE,OAAO;YACX,CAAC;YAED,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;YAE1D,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,IAAI,UAAU,GAAG,CAAC,CAAC;YAEnB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,cAAc,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC;gBACvD,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBAClB,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;oBACrE,OAAO;gBACX,CAAC;gBACD,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;oBACX,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;gBACxD,CAAC;gBAED,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACtB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACzC,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC;gBACxC,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;gBAChC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;oBAC1B,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,SAAS,UAAU,GAAG,CAAC,KAAK,KAAK,EAAE,CAAC,CAAC;gBAC1F,CAAC;gBACD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAChB,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC;gBAE9E,UAAU,IAAI,CAAC,CAAC;gBAChB,UAAU,IAAI,CAAC,CAAC;gBAChB,YAAY,IAAI,KAAK,CAAC;gBACtB,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;YAClE,CAAC;YAED,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;gBACnB,MAAM,IAAI,KAAK,CAAC,SAAS,KAAK,wBAAwB,CAAC,CAAC;YAC5D,CAAC;YAED,QAAQ,GAAG,YAAY,GAAG,UAAU,CAAC;YACrC,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;YAE1E,MAAM,gBAAgB,GAAG,KAAK,GAAG,IAAI,CAAC,eAAe,KAAK,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,MAAM,CAAC;YACrF,IAAI,gBAAgB,EAAE,CAAC;gBACnB,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;gBACvC,MAAM;oBACF,IAAI,EAAE,YAAY;oBAClB,KAAK;oBACL,KAAK;oBACL,QAAQ,EAAE,iBAAiB,CAAC,KAAK,CAAC;iBACrC,CAAC;YACN,CAAC;QACL,CAAC;QAED,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,aAAa,EAAE,QAAQ,EAAE,CAAC;IAC5F,CAAC;IAED,yFAAyF;IACzF,KAAK,CAAC,GAAG,CAAC,IAA+B;QACrC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,IAAI,aAAiC,CAAC;QACtC,IAAI,cAAyC,CAAC;QAE9C,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YAC3C,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,OAAO;oBACR,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC;oBACnB,MAAM;gBACV,KAAK,WAAW;oBACZ,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC;oBACrB,aAAa,GAAG,KAAK,CAAC,QAAQ,CAAC;oBAC/B,MAAM;gBACV,KAAK,YAAY;oBACb,cAAc,GAAG,KAAK,CAAC,QAAQ,CAAC;oBAChC,MAAM;gBACV,KAAK,SAAS;oBACV,OAAO,GAAG,IAAI,CAAC;oBACf,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC;oBACrB,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC;oBACnB,MAAM;gBACV,KAAK,MAAM;oBACP,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;oBACtB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;oBACpB,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC;oBACpC,MAAM;gBACV;oBACI,MAAM;YACd,CAAC;QACL,CAAC;QAED,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,aAAa,EAAE,cAAc,EAAE,CAAC;IACrE,CAAC;CACJ"}

package/package.json

@@ -1,9 +1,9 @@
 {
     "name": "@dxo/train",
-    "version": "0.0.8",
+    "version": "0.0.10",
     "type": "module",
-    "description": "DXO training loop — fitIter / TrainEvent / checkpoint (developer preview, API unstable)",
-    "license": "MIT OR Apache-2.0",
+    "description": "DXO training loop - fitIter / TrainEvent / checkpoint (developer preview, API unstable)",
+    "license": "Apache-2.0",
     "main": "./dist/index.js",
     "types": "./dist/index.d.ts",
     "exports": {
@@ -20,23 +20,29 @@
         "build": "tsc -p tsconfig.json"
     },
     "dependencies": {
-        "@dxo/core": "0.0.8",
-        "@dxo/data": "0.0.8",
-        "@dxo/nn": "0.0.8",
-        "@dxo/optimizer": "0.0.8",
-        "@dxo/serialize": "0.0.8"
+        "@dxo/core": "0.0.10",
+        "@dxo/data": "0.0.10",
+        "@dxo/nn": "0.0.10",
+        "@dxo/optimizer": "0.0.10",
+        "@dxo/serialize": "0.0.10"
     },
     "keywords": [
         "dxo",
         "trainer",
         "training",
-        "typescript"
+        "typescript",
+        "deep-learning"
     ],
-    "publishConfig": {
-        "access": "public"
-    },
+    "homepage": "https://github.com/ai4waifu/dxo-framework#readme",
     "repository": {
         "type": "git",
-        "url": "git+https://github.com/ai4waifu/dxo-framework.git"
+        "url": "https://github.com/ai4waifu/dxo-framework.git",
+        "directory": "projects/runtimes/dxo-train"
+    },
+    "bugs": {
+        "url": "https://github.com/ai4waifu/dxo-framework/issues"
+    },
+    "publishConfig": {
+        "access": "public"
     }
 }