@dxo/graph 0.0.0 → 0.0.9

package/README.md

@@ -1,3 +1,5 @@
 # @dxo/graph
 
-Placeholder package (0.0.0). Reserved for the DXO project.
+**Placeholder — developer preview, API unstable.**
+
+Serializable model-graph IR for Studio, CLI, and exporters. Distinguishes **module** / **execution** / **kernel** views. Not a Netron replacement yet; no full format parsers in this stub.

package/dist/index.d.ts

@@ -0,0 +1,34 @@
+/**
+ * UI-agnostic model graph IR placeholder (Living `10` / `@dxo/graph`).
+ * Module graph ≠ execution trace ≠ kernel graph — do not collapse them.
+ */
+export declare const MODEL_GRAPH_FORMAT: "dxo-model-graph";
+export declare const MODEL_GRAPH_VERSION: 0;
+export type GraphViewKind = 'module' | 'execution' | 'kernel';
+export type GraphNode = {
+    id: string;
+    kind: string;
+    label?: string;
+    modulePath?: string;
+    attrs?: Record<string, unknown>;
+};
+export type GraphEdge = {
+    id: string;
+    source: string;
+    target: string;
+    tensor?: {
+        shape?: number[];
+        dtype?: string;
+        device?: string;
+    };
+};
+export type ModelGraphV0 = {
+    format: typeof MODEL_GRAPH_FORMAT;
+    version: typeof MODEL_GRAPH_VERSION;
+    view: GraphViewKind;
+    nodes: GraphNode[];
+    edges: GraphEdge[];
+};
+export declare function graphVersion(): string;
+export { emptyModelGraph, moduleGraphFromLinear, moduleGraphFromModule, moduleGraphFromSequential, serializeModelGraph, } from './module-graph.js';
+//# 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;;;GAGG;AAEH,eAAO,MAAM,kBAAkB,EAAG,iBAA0B,CAAC;AAC7D,eAAO,MAAM,mBAAmB,EAAG,CAAU,CAAC;AAE9C,MAAM,MAAM,aAAa,GAAG,QAAQ,GAAG,WAAW,GAAG,QAAQ,CAAC;AAE9D,MAAM,MAAM,SAAS,GAAG;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC,CAAC;AAEF,MAAM,MAAM,SAAS,GAAG;IACpB,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,EAAE,CAAC;QAAC,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;CAClE,CAAC;AAEF,MAAM,MAAM,YAAY,GAAG;IACvB,MAAM,EAAE,OAAO,kBAAkB,CAAC;IAClC,OAAO,EAAE,OAAO,mBAAmB,CAAC;IACpC,IAAI,EAAE,aAAa,CAAC;IACpB,KAAK,EAAE,SAAS,EAAE,CAAC;IACnB,KAAK,EAAE,SAAS,EAAE,CAAC;CACtB,CAAC;AAEF,wBAAgB,YAAY,IAAI,MAAM,CAErC;AAED,OAAO,EACH,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,yBAAyB,EACzB,mBAAmB,GACtB,MAAM,mBAAmB,CAAC"}

package/dist/index.js

@@ -0,0 +1,11 @@
+/**
+ * UI-agnostic model graph IR placeholder (Living `10` / `@dxo/graph`).
+ * Module graph ≠ execution trace ≠ kernel graph — do not collapse them.
+ */
+export const MODEL_GRAPH_FORMAT = 'dxo-model-graph';
+export const MODEL_GRAPH_VERSION = 0;
+export function graphVersion() {
+    return `${MODEL_GRAPH_FORMAT}@${MODEL_GRAPH_VERSION}`;
+}
+export { emptyModelGraph, moduleGraphFromLinear, moduleGraphFromModule, moduleGraphFromSequential, serializeModelGraph, } from './module-graph.js';
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,MAAM,CAAC,MAAM,kBAAkB,GAAG,iBAA0B,CAAC;AAC7D,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAU,CAAC;AA2B9C,MAAM,UAAU,YAAY;IACxB,OAAO,GAAG,kBAAkB,IAAI,mBAAmB,EAAE,CAAC;AAC1D,CAAC;AAED,OAAO,EACH,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,yBAAyB,EACzB,mBAAmB,GACtB,MAAM,mBAAmB,CAAC"}

package/dist/module-graph.d.ts

@@ -0,0 +1,10 @@
+import type { Module } from '@dxo/nn';
+import { Linear, Sequential } from '@dxo/nn';
+import { type GraphViewKind, type ModelGraphV0 } from './index.js';
+export declare function emptyModelGraph(view?: GraphViewKind): ModelGraphV0;
+export declare function serializeModelGraph(graph: ModelGraphV0): string;
+export declare function moduleGraphFromLinear(linear: Linear, path?: string): ModelGraphV0;
+export declare function moduleGraphFromSequential(seq: Sequential, path?: string): ModelGraphV0;
+/** Build a module graph from any @dxo/nn Module (Linear or Sequential supported). */
+export declare function moduleGraphFromModule(mod: Module, path?: string): ModelGraphV0;
+//# sourceMappingURL=module-graph.d.ts.map

package/dist/module-graph.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"module-graph.d.ts","sourceRoot":"","sources":["../src/module-graph.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAAE,MAAM,EAAQ,UAAU,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAkC,KAAK,aAAa,EAA2C,KAAK,YAAY,EAAE,MAAM,YAAY,CAAC;AAE5I,wBAAgB,eAAe,CAAC,IAAI,GAAE,aAAwB,GAAG,YAAY,CAQ5E;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,YAAY,GAAG,MAAM,CAE/D;AAUD,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,SAAW,GAAG,YAAY,CAenF;AAED,wBAAgB,yBAAyB,CAAC,GAAG,EAAE,UAAU,EAAE,IAAI,SAAe,GAAG,YAAY,CA2B5F;AAED,qFAAqF;AACrF,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,SAAU,GAAG,YAAY,CAI/E"}

package/dist/module-graph.js

@@ -0,0 +1,70 @@
+import { Linear, Relu, Sequential } from '@dxo/nn';
+import { MODEL_GRAPH_FORMAT, MODEL_GRAPH_VERSION } from './index.js';
+export function emptyModelGraph(view = 'module') {
+    return {
+        format: MODEL_GRAPH_FORMAT,
+        version: MODEL_GRAPH_VERSION,
+        view,
+        nodes: [],
+        edges: [],
+    };
+}
+export function serializeModelGraph(graph) {
+    return JSON.stringify(graph);
+}
+function node(id, kind, label, modulePath, attrs) {
+    return { id, kind, label, modulePath, attrs };
+}
+function edge(id, source, target, tensor) {
+    return { id, source, target, tensor };
+}
+export function moduleGraphFromLinear(linear, path = 'linear') {
+    const nodes = [
+        node('input', 'tensor', 'input'),
+        node('linear', 'Linear', 'Linear', path, {
+            inFeatures: linear.inFeatures,
+            outFeatures: linear.outFeatures,
+            params: ['weight', 'bias'],
+        }),
+        node('output', 'tensor', 'output'),
+    ];
+    const edges = [
+        edge('e-in-linear', 'input', 'linear', { dtype: 'f32' }),
+        edge('e-linear-out', 'linear', 'output', { dtype: 'f32' }),
+    ];
+    return { format: MODEL_GRAPH_FORMAT, version: MODEL_GRAPH_VERSION, view: 'module', nodes, edges };
+}
+export function moduleGraphFromSequential(seq, path = 'sequential') {
+    const nodes = [node('input', 'tensor', 'input')];
+    const edges = [];
+    let prev = 'input';
+    seq.layers.forEach((layer, i) => {
+        const id = `layer_${i}`;
+        if (layer instanceof Linear) {
+            nodes.push(node(id, 'Linear', `Linear(${layer.inFeatures}->${layer.outFeatures})`, `${path}/${id}`, {
+                inFeatures: layer.inFeatures,
+                outFeatures: layer.outFeatures,
+            }));
+        }
+        else if (layer instanceof Relu) {
+            nodes.push(node(id, 'Relu', 'Relu', `${path}/${id}`));
+        }
+        else {
+            nodes.push(node(id, layer.constructor.name, layer.constructor.name, `${path}/${id}`));
+        }
+        edges.push(edge(`e-${prev}-${id}`, prev, id, { dtype: 'f32' }));
+        prev = id;
+    });
+    nodes.push(node('output', 'tensor', 'output'));
+    edges.push(edge(`e-${prev}-output`, prev, 'output', { dtype: 'f32' }));
+    return { format: MODEL_GRAPH_FORMAT, version: MODEL_GRAPH_VERSION, view: 'module', nodes, edges };
+}
+/** Build a module graph from any @dxo/nn Module (Linear or Sequential supported). */
+export function moduleGraphFromModule(mod, path = 'model') {
+    if (mod instanceof Sequential)
+        return moduleGraphFromSequential(mod, path);
+    if (mod instanceof Linear)
+        return moduleGraphFromLinear(mod, path);
+    throw new Error(`moduleGraphFromModule: unsupported module ${mod.constructor.name}`);
+}
+//# sourceMappingURL=module-graph.js.map

package/dist/module-graph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"module-graph.js","sourceRoot":"","sources":["../src/module-graph.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAsD,kBAAkB,EAAE,mBAAmB,EAAqB,MAAM,YAAY,CAAC;AAE5I,MAAM,UAAU,eAAe,CAAC,OAAsB,QAAQ;IAC1D,OAAO;QACH,MAAM,EAAE,kBAAkB;QAC1B,OAAO,EAAE,mBAAmB;QAC5B,IAAI;QACJ,KAAK,EAAE,EAAE;QACT,KAAK,EAAE,EAAE;KACZ,CAAC;AACN,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAmB;IACnD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AACjC,CAAC;AAED,SAAS,IAAI,CAAC,EAAU,EAAE,IAAY,EAAE,KAAa,EAAE,UAAmB,EAAE,KAA+B;IACvG,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;AAClD,CAAC;AAED,SAAS,IAAI,CAAC,EAAU,EAAE,MAAc,EAAE,MAAc,EAAE,MAA4B;IAClF,OAAO,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;AAC1C,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,MAAc,EAAE,IAAI,GAAG,QAAQ;IACjE,MAAM,KAAK,GAAgB;QACvB,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC;QAChC,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,IAAI,EAAE;YACrC,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,WAAW,EAAE,MAAM,CAAC,WAAW;YAC/B,MAAM,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC;SAC7B,CAAC;QACF,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC;KACrC,CAAC;IACF,MAAM,KAAK,GAAgB;QACvB,IAAI,CAAC,aAAa,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;QACxD,IAAI,CAAC,cAAc,EAAE,QAAQ,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;KAC7D,CAAC;IACF,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;AACtG,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,GAAe,EAAE,IAAI,GAAG,YAAY;IAC1E,MAAM,KAAK,GAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;IAC9D,MAAM,KAAK,GAAgB,EAAE,CAAC;IAC9B,IAAI,IAAI,GAAG,OAAO,CAAC;IAEnB,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;QAC5B,MAAM,EAAE,GAAG,SAAS,CAAC,EAAE,CAAC;QACxB,IAAI,KAAK,YAAY,MAAM,EAAE,CAAC;YAC1B,KAAK,CAAC,IAAI,CACN,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,UAAU,KAAK,CAAC,UAAU,KAAK,KAAK,CAAC,WAAW,GAAG,EAAE,GAAG,IAAI,IAAI,EAAE,EAAE,EAAE;gBACrF,UAAU,EAAE,KAAK,CAAC,UAAU;gBAC5B,WAAW,EAAE,KAAK,CAAC,WAAW;aACjC,CAAC,CACL,CAAC;QACN,CAAC;aAAM,IAAI,KAAK,YAAY,IAAI,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1D,CAAC;aAAM,CAAC;YACJ,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,KAAK,CAAC,WAAW,CAAC,IAAI,EAAE,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1F,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;QAChE,IAAI,GAAG,EAAE,CAAC;IACd,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC/C,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,IAAI,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;IAEvE,OAAO,EAAE,MAAM,EAAE,kBAAkB,EAAE,OAAO,EAAE,mBAAmB,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC;AACtG,CAAC;AAED,qFAAqF;AACrF,MAAM,UAAU,qBAAqB,CAAC,GAAW,EAAE,IAAI,GAAG,OAAO;IAC7D,IAAI,GAAG,YAAY,UAAU;QAAE,OAAO,yBAAyB,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IAC3E,IAAI,GAAG,YAAY,MAAM;QAAE,OAAO,qBAAqB,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACnE,MAAM,IAAI,KAAK,CAAC,6CAA6C,GAAG,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,CAAC;AACzF,CAAC"}

package/package.json

@@ -1,10 +1,39 @@
 {
-  "name": "@dxo/graph",
-  "version": "0.0.0",
-  "description": "DXO placeholder — not for production use.",
-  "license": "MIT",
-  "private": false,
-  "files": [
-    "README.md"
-  ]
+    "name": "@dxo/graph",
+    "version": "0.0.9",
+    "type": "module",
+    "description": "DXO model graph IR — module / execution / kernel views (placeholder, API unstable)",
+    "license": "MIT OR Apache-2.0",
+    "main": "./dist/index.js",
+    "types": "./dist/index.d.ts",
+    "exports": {
+        ".": {
+            "types": "./dist/index.d.ts",
+            "default": "./dist/index.js"
+        }
+    },
+    "files": [
+        "dist",
+        "README.md"
+    ],
+    "scripts": {
+        "build": "tsc -p tsconfig.json"
+    },
+    "dependencies": {
+        "@dxo/nn": "0.0.9"
+    },
+    "keywords": [
+        "dxo",
+        "graph",
+        "model",
+        "netron",
+        "typescript"
+    ],
+    "publishConfig": {
+        "access": "public"
+    },
+    "repository": {
+        "type": "git",
+        "url": "git+https://github.com/ai4waifu/dxo-framework.git"
+    }
 }