@elaraai/east-py-datascience 0.0.2-beta.1 → 0.0.2-beta.5

package/dist/torch/torch.d.ts

@@ -258,6 +258,202 @@ export declare const torch_mlp_predict: import("@elaraai/east").PlatformDefiniti
         output_dim: IntegerType;
     }>;
 }>, ArrayType<ArrayType<FloatType>>], ArrayType<FloatType>>;
+/**
+ * Train a PyTorch MLP model with multi-output support.
+ *
+ * Supports multi-output regression (predicting multiple values per sample)
+ * and autoencoders (where input equals target for reconstruction learning).
+ * Output dimension is inferred from y.shape[1] unless overridden in config.
+ *
+ * @param X - Feature matrix (n_samples x n_features)
+ * @param y - Target matrix (n_samples x n_outputs)
+ * @param mlp_config - MLP architecture configuration
+ * @param train_config - Training configuration
+ * @returns Model blob and training result
+ */
+export declare const torch_mlp_train_multi: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
+    /** Hidden layer sizes, e.g., [64, 32] */
+    hidden_layers: ArrayType<IntegerType>;
+    /** Activation function (default relu) */
+    activation: OptionType<VariantType<{
+        /** Rectified Linear Unit */
+        relu: StructType<{}>;
+        /** Hyperbolic tangent */
+        tanh: StructType<{}>;
+        /** Sigmoid function */
+        sigmoid: StructType<{}>;
+        /** Leaky ReLU */
+        leaky_relu: StructType<{}>;
+    }>>;
+    /** Dropout rate (default 0.0) */
+    dropout: OptionType<FloatType>;
+    /** Output dimension (default 1) */
+    output_dim: OptionType<IntegerType>;
+}>, StructType<{
+    /** Number of epochs (default 100) */
+    epochs: OptionType<IntegerType>;
+    /** Batch size (default 32) */
+    batch_size: OptionType<IntegerType>;
+    /** Learning rate (default 0.001) */
+    learning_rate: OptionType<FloatType>;
+    /** Loss function (default mse) */
+    loss: OptionType<VariantType<{
+        /** Mean Squared Error (regression) */
+        mse: StructType<{}>;
+        /** Mean Absolute Error (regression) */
+        mae: StructType<{}>;
+        /** Cross Entropy (classification) */
+        cross_entropy: StructType<{}>;
+    }>>;
+    /** Optimizer (default adam) */
+    optimizer: OptionType<VariantType<{
+        /** Adam optimizer */
+        adam: StructType<{}>;
+        /** Stochastic Gradient Descent */
+        sgd: StructType<{}>;
+        /** AdamW with weight decay */
+        adamw: StructType<{}>;
+        /** RMSprop optimizer */
+        rmsprop: StructType<{}>;
+    }>>;
+    /** Early stopping patience, 0 = disabled */
+    early_stopping: OptionType<IntegerType>;
+    /** Validation split fraction (default 0.2) */
+    validation_split: OptionType<FloatType>;
+    /** Random seed for reproducibility */
+    random_state: OptionType<IntegerType>;
+}>], StructType<{
+    /** Trained model blob */
+    model: VariantType<{
+        torch_mlp: StructType<{
+            data: BlobType;
+            n_features: IntegerType;
+            hidden_layers: ArrayType<IntegerType>;
+            output_dim: IntegerType;
+        }>;
+    }>;
+    /** Training result with losses */
+    result: StructType<{
+        /** Training loss per epoch */
+        train_losses: ArrayType<FloatType>;
+        /** Validation loss per epoch */
+        val_losses: ArrayType<FloatType>;
+        /** Best epoch (for early stopping) */
+        best_epoch: IntegerType;
+    }>;
+}>>;
+/**
+ * Make predictions with a trained PyTorch MLP (multi-output).
+ *
+ * Returns a matrix where each row contains the predicted outputs for a sample.
+ *
+ * @param model - Trained MLP model blob
+ * @param X - Feature matrix (n_samples x n_features)
+ * @returns Predicted matrix (n_samples x n_outputs)
+ */
+export declare const torch_mlp_predict_multi: import("@elaraai/east").PlatformDefinition<[VariantType<{
+    /** PyTorch MLP model */
+    torch_mlp: StructType<{
+        /** Cloudpickle serialized model */
+        data: BlobType;
+        /** Number of input features */
+        n_features: IntegerType;
+        /** Hidden layer sizes */
+        hidden_layers: ArrayType<IntegerType>;
+        /** Output dimension */
+        output_dim: IntegerType;
+    }>;
+}>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
+/**
+ * Extract intermediate layer activations (embeddings) from a trained MLP.
+ *
+ * For autoencoders, this allows extracting the bottleneck representation.
+ * The layer_index specifies which hidden layer's output to return (0-indexed).
+ *
+ * For an autoencoder with architecture [input -> 8 -> 2 -> 8 -> output]
+ * (hidden_layers: [8, 2, 8]):
+ * - layer_index=0: output after first hidden layer (8 features)
+ * - layer_index=1: output after second hidden layer (2 features) <- bottleneck
+ * - layer_index=2: output after third hidden layer (8 features)
+ *
+ * @param model - Trained MLP model blob
+ * @param X - Feature matrix (n_samples x n_features)
+ * @param layer_index - Which hidden layer's output to return (0-indexed)
+ * @returns Embedding matrix (n_samples x hidden_dim at that layer)
+ *
+ * @example
+ * ```ts
+ * // Train autoencoder: 4 features -> 8 -> 2 (bottleneck) -> 8 -> 4 features
+ * const mlp_config = $.let({
+ *     hidden_layers: [8n, 2n, 8n],
+ *     activation: variant('some', variant('relu', {})),
+ *     dropout: variant('none', null),
+ *     output_dim: variant('none', null),
+ * });
+ * const output = $.let(Torch.mlpTrainMulti(X, X, mlp_config, train_config));
+ *
+ * // Extract bottleneck embeddings (layer_index=1 for the 2-dim bottleneck)
+ * const embeddings = $.let(Torch.mlpEncode(output.model, X, 1n));
+ * // embeddings is now (n_samples x 2)
+ * ```
+ */
+export declare const torch_mlp_encode: import("@elaraai/east").PlatformDefinition<[VariantType<{
+    /** PyTorch MLP model */
+    torch_mlp: StructType<{
+        /** Cloudpickle serialized model */
+        data: BlobType;
+        /** Number of input features */
+        n_features: IntegerType;
+        /** Hidden layer sizes */
+        hidden_layers: ArrayType<IntegerType>;
+        /** Output dimension */
+        output_dim: IntegerType;
+    }>;
+}>, ArrayType<ArrayType<FloatType>>, IntegerType], ArrayType<ArrayType<FloatType>>>;
+/**
+ * Decode embeddings back through the decoder portion of an MLP.
+ *
+ * For autoencoders, this takes bottleneck activations and runs them through
+ * the decoder to reconstruct the output. This is the complement to mlpEncode.
+ *
+ * For an autoencoder with architecture [input -> 8 -> 2 -> 8 -> output]
+ * (hidden_layers: [8, 2, 8]):
+ * - layer_index=1: Start from the 2-dim bottleneck, run through layers 2+ to output
+ * - layer_index=0: Start from the 8-dim first layer, run through layers 1+ to output
+ *
+ * Use case: Compute weighted average of origin embeddings, then decode to
+ * get the reconstructed blend weight distribution.
+ *
+ * @param model - Trained MLP model blob
+ * @param embeddings - Embedding matrix (n_samples x hidden_dim at layer_index)
+ * @param layer_index - Which hidden layer the embeddings come from (0-indexed)
+ * @returns Decoded output matrix (n_samples x output_dim)
+ *
+ * @example
+ * ```ts
+ * // After training autoencoder and extracting embeddings...
+ * const origin_embeddings = $.let(Torch.mlpEncode(output.model, X_onehot, 1n));
+ *
+ * // Compute weighted blend embedding (e.g., 50% origin A + 50% origin B)
+ * const blend_embedding = $.let(...); // weighted average of origin embeddings
+ *
+ * // Decode back to weight distribution
+ * const reconstructed = $.let(Torch.mlpDecode(output.model, blend_embedding, 1n));
+ * ```
+ */
+export declare const torch_mlp_decode: import("@elaraai/east").PlatformDefinition<[VariantType<{
+    /** PyTorch MLP model */
+    torch_mlp: StructType<{
+        /** Cloudpickle serialized model */
+        data: BlobType;
+        /** Number of input features */
+        n_features: IntegerType;
+        /** Hidden layer sizes */
+        hidden_layers: ArrayType<IntegerType>;
+        /** Output dimension */
+        output_dim: IntegerType;
+    }>;
+}>, ArrayType<ArrayType<FloatType>>, IntegerType], ArrayType<ArrayType<FloatType>>>;
 /**
  * Type definitions for PyTorch functions.
  */
@@ -431,7 +627,7 @@ export declare const TorchTypes: {
  * ```
  */
 export declare const Torch: {
-    /** Train MLP model */
+    /** Train MLP model (single output) */
     readonly mlpTrain: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<FloatType>, StructType<{
         /** Hidden layer sizes, e.g., [64, 32] */
         hidden_layers: ArrayType<IntegerType>;
@@ -503,7 +699,7 @@ export declare const Torch: {
             best_epoch: IntegerType;
         }>;
     }>>;
-    /** Make predictions with MLP */
+    /** Make predictions with MLP (single output) */
     readonly mlpPredict: import("@elaraai/east").PlatformDefinition<[VariantType<{
         /** PyTorch MLP model */
         torch_mlp: StructType<{
@@ -517,6 +713,120 @@ export declare const Torch: {
             output_dim: IntegerType;
         }>;
     }>, ArrayType<ArrayType<FloatType>>], ArrayType<FloatType>>;
+    /** Train MLP model (multi-output) */
+    readonly mlpTrainMulti: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
+        /** Hidden layer sizes, e.g., [64, 32] */
+        hidden_layers: ArrayType<IntegerType>;
+        /** Activation function (default relu) */
+        activation: OptionType<VariantType<{
+            /** Rectified Linear Unit */
+            relu: StructType<{}>;
+            /** Hyperbolic tangent */
+            tanh: StructType<{}>;
+            /** Sigmoid function */
+            sigmoid: StructType<{}>;
+            /** Leaky ReLU */
+            leaky_relu: StructType<{}>;
+        }>>;
+        /** Dropout rate (default 0.0) */
+        dropout: OptionType<FloatType>;
+        /** Output dimension (default 1) */
+        output_dim: OptionType<IntegerType>;
+    }>, StructType<{
+        /** Number of epochs (default 100) */
+        epochs: OptionType<IntegerType>;
+        /** Batch size (default 32) */
+        batch_size: OptionType<IntegerType>;
+        /** Learning rate (default 0.001) */
+        learning_rate: OptionType<FloatType>;
+        /** Loss function (default mse) */
+        loss: OptionType<VariantType<{
+            /** Mean Squared Error (regression) */
+            mse: StructType<{}>;
+            /** Mean Absolute Error (regression) */
+            mae: StructType<{}>;
+            /** Cross Entropy (classification) */
+            cross_entropy: StructType<{}>;
+        }>>;
+        /** Optimizer (default adam) */
+        optimizer: OptionType<VariantType<{
+            /** Adam optimizer */
+            adam: StructType<{}>;
+            /** Stochastic Gradient Descent */
+            sgd: StructType<{}>;
+            /** AdamW with weight decay */
+            adamw: StructType<{}>;
+            /** RMSprop optimizer */
+            rmsprop: StructType<{}>;
+        }>>;
+        /** Early stopping patience, 0 = disabled */
+        early_stopping: OptionType<IntegerType>;
+        /** Validation split fraction (default 0.2) */
+        validation_split: OptionType<FloatType>;
+        /** Random seed for reproducibility */
+        random_state: OptionType<IntegerType>;
+    }>], StructType<{
+        /** Trained model blob */
+        model: VariantType<{
+            torch_mlp: StructType<{
+                data: BlobType;
+                n_features: IntegerType;
+                hidden_layers: ArrayType<IntegerType>;
+                output_dim: IntegerType;
+            }>;
+        }>;
+        /** Training result with losses */
+        result: StructType<{
+            /** Training loss per epoch */
+            train_losses: ArrayType<FloatType>;
+            /** Validation loss per epoch */
+            val_losses: ArrayType<FloatType>;
+            /** Best epoch (for early stopping) */
+            best_epoch: IntegerType;
+        }>;
+    }>>;
+    /** Make predictions with MLP (multi-output) */
+    readonly mlpPredictMulti: import("@elaraai/east").PlatformDefinition<[VariantType<{
+        /** PyTorch MLP model */
+        torch_mlp: StructType<{
+            /** Cloudpickle serialized model */
+            data: BlobType;
+            /** Number of input features */
+            n_features: IntegerType;
+            /** Hidden layer sizes */
+            hidden_layers: ArrayType<IntegerType>;
+            /** Output dimension */
+            output_dim: IntegerType;
+        }>;
+    }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
+    /** Extract intermediate layer activations (embeddings) from MLP */
+    readonly mlpEncode: import("@elaraai/east").PlatformDefinition<[VariantType<{
+        /** PyTorch MLP model */
+        torch_mlp: StructType<{
+            /** Cloudpickle serialized model */
+            data: BlobType;
+            /** Number of input features */
+            n_features: IntegerType;
+            /** Hidden layer sizes */
+            hidden_layers: ArrayType<IntegerType>;
+            /** Output dimension */
+            output_dim: IntegerType;
+        }>;
+    }>, ArrayType<ArrayType<FloatType>>, IntegerType], ArrayType<ArrayType<FloatType>>>;
+    /** Decode embeddings back through decoder portion of MLP */
+    readonly mlpDecode: import("@elaraai/east").PlatformDefinition<[VariantType<{
+        /** PyTorch MLP model */
+        torch_mlp: StructType<{
+            /** Cloudpickle serialized model */
+            data: BlobType;
+            /** Number of input features */
+            n_features: IntegerType;
+            /** Hidden layer sizes */
+            hidden_layers: ArrayType<IntegerType>;
+            /** Output dimension */
+            output_dim: IntegerType;
+        }>;
+    }>, ArrayType<ArrayType<FloatType>>, IntegerType], ArrayType<ArrayType<FloatType>>>;
     /** Type definitions */
     readonly Types: {
         /** Vector type (array of floats) */

package/dist/torch/torch.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"torch.d.ts","sourceRoot":"","sources":["../../src/torch/torch.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACZ,MAAM,eAAe,CAAC;AAIvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMrD;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,4BAA4B;;IAE5B,yBAAyB;;IAEzB,uBAAuB;;IAEvB,iBAAiB;;EAEnB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,aAAa;IACtB,sCAAsC;;IAEtC,uCAAuC;;IAEvC,qCAAqC;;EAEvC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,kBAAkB;IAC3B,qBAAqB;;IAErB,kCAAkC;;IAElC,8BAA8B;;IAE9B,wBAAwB;;EAE1B,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,kBAAkB;IAC3B,yCAAyC;;IAEzC,yCAAyC;;QA9CzC,4BAA4B;;QAE5B,yBAAyB;;QAEzB,uBAAuB;;QAEvB,iBAAiB;;;IA0CjB,iCAAiC;;IAEjC,mCAAmC;;EAErC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAC7B,qCAAqC;;IAErC,8BAA8B;;IAE9B,oCAAoC;;IAEpC,kCAAkC;;QAlDlC,sCAAsC;;QAEtC,uCAAuC;;QAEvC,qCAAqC;;;IAgDrC,+BAA+B;;QAxC/B,qBAAqB;;QAErB,kCAAkC;;QAElC,8BAA8B;;QAE9B,wBAAwB;;;IAoCxB,4CAA4C;;IAE5C,8CAA8C;;IAE9C,sCAAsC;;EAExC,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAC7B,8BAA8B;;IAE9B,gCAAgC;;IAEhC,sCAAsC;;EAExC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAC7B,yBAAyB;;;;;;;;;IASzB,kCAAkC;;QArBlC,8BAA8B;;QAE9B,gCAAgC;;QAEhC,sCAAsC;;;EAmBxC,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,kBAAkB;IAC3B,wBAAwB;;QAEpB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,yBAAyB;;QAEzB,uBAAuB;;;EAG7B,CAAC;AAMH;;;;;;;;GAQG;AACH,eAAO,MAAM,eAAe;IAnGxB,yCAAyC;;IAEzC,yCAAyC;;QA9CzC,4BAA4B;;QAE5B,yBAAyB;;QAEzB,uBAAuB;;QAEvB,iBAAiB;;;IA0CjB,iCAAiC;;IAEjC,mCAAmC;;;IAQnC,qCAAqC;;IAErC,8BAA8B;;IAE9B,oCAAoC;;IAEpC,kCAAkC;;QAlDlC,sCAAsC;;QAEtC,uCAAuC;;QAEvC,qCAAqC;;;IAgDrC,+BAA+B;;QAxC/B,qBAAqB;;QAErB,kCAAkC;;QAElC,8BAA8B;;QAE9B,wBAAwB;;;IAoCxB,4CAA4C;;IAE5C,8CAA8C;;IAE9C,sCAAsC;;;IAwBtC,yBAAyB;;;;;;;;;IASzB,kCAAkC;;QArBlC,8BAA8B;;QAE9B,gCAAgC;;QAEhC,sCAAsC;;;GA2DzC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,iBAAiB;IAvC1B,wBAAwB;;QAEpB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,yBAAyB;;QAEzB,uBAAuB;;;2DAmC9B,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,UAAU;IACnB,oCAAoC;;IAEpC,uCAAuC;;IAEvC,+BAA+B;;QA9K/B,4BAA4B;;QAE5B,yBAAyB;;QAEzB,uBAAuB;;QAEvB,iBAAiB;;;IA0KjB,yBAAyB;;QAlKzB,sCAAsC;;QAEtC,uCAAuC;;QAEvC,qCAAqC;;;IAgKrC,qBAAqB;;QAxJrB,qBAAqB;;QAErB,kCAAkC;;QAElC,8BAA8B;;QAE9B,wBAAwB;;;IAoJxB,6BAA6B;;QAxI7B,yCAAyC;;QAEzC,yCAAyC;;YA9CzC,4BAA4B;;YAE5B,yBAAyB;;YAEzB,uBAAuB;;YAEvB,iBAAiB;;;QA0CjB,iCAAiC;;QAEjC,mCAAmC;;;IAoInC,kCAAkC;;QA5HlC,qCAAqC;;QAErC,8BAA8B;;QAE9B,oCAAoC;;QAEpC,kCAAkC;;YAlDlC,sCAAsC;;YAEtC,uCAAuC;;YAEvC,qCAAqC;;;QAgDrC,+BAA+B;;YAxC/B,qBAAqB;;YAErB,kCAAkC;;YAElC,8BAA8B;;YAE9B,wBAAwB;;;QAoCxB,4CAA4C;;QAE5C,8CAA8C;;QAE9C,sCAAsC;;;IAgHtC,2BAA2B;;QApG3B,8BAA8B;;QAE9B,gCAAgC;;QAEhC,sCAAsC;;;IAkGtC,4CAA4C;;QA1F5C,yBAAyB;;;;;;;;;QASzB,kCAAkC;;YArBlC,8BAA8B;;YAE9B,gCAAgC;;YAEhC,sCAAsC;;;;IAoGtC,yCAAyC;;QAvEzC,wBAAwB;;YAEpB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,yBAAyB;;YAEzB,uBAAuB;;;;CAiErB,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,eAAO,MAAM,KAAK;IACd,sBAAsB;;QAtLtB,yCAAyC;;QAEzC,yCAAyC;;YA9CzC,4BAA4B;;YAE5B,yBAAyB;;YAEzB,uBAAuB;;YAEvB,iBAAiB;;;QA0CjB,iCAAiC;;QAEjC,mCAAmC;;;QAQnC,qCAAqC;;QAErC,8BAA8B;;QAE9B,oCAAoC;;QAEpC,kCAAkC;;YAlDlC,sCAAsC;;YAEtC,uCAAuC;;YAEvC,qCAAqC;;;QAgDrC,+BAA+B;;YAxC/B,qBAAqB;;YAErB,kCAAkC;;YAElC,8BAA8B;;YAE9B,wBAAwB;;;QAoCxB,4CAA4C;;QAE5C,8CAA8C;;QAE9C,sCAAsC;;;QAwBtC,yBAAyB;;;;;;;;;QASzB,kCAAkC;;YArBlC,8BAA8B;;YAE9B,gCAAgC;;YAEhC,sCAAsC;;;;IA4ItC,gCAAgC;;QA/GhC,wBAAwB;;YAEpB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,yBAAyB;;YAEzB,uBAAuB;;;;IAyG3B,uBAAuB;;QA5DvB,oCAAoC;;QAEpC,uCAAuC;;QAEvC,+BAA+B;;YA9K/B,4BAA4B;;YAE5B,yBAAyB;;YAEzB,uBAAuB;;YAEvB,iBAAiB;;;QA0KjB,yBAAyB;;YAlKzB,sCAAsC;;YAEtC,uCAAuC;;YAEvC,qCAAqC;;;QAgKrC,qBAAqB;;YAxJrB,qBAAqB;;YAErB,kCAAkC;;YAElC,8BAA8B;;YAE9B,wBAAwB;;;QAoJxB,6BAA6B;;YAxI7B,yCAAyC;;YAEzC,yCAAyC;;gBA9CzC,4BAA4B;;gBAE5B,yBAAyB;;gBAEzB,uBAAuB;;gBAEvB,iBAAiB;;;YA0CjB,iCAAiC;;YAEjC,mCAAmC;;;QAoInC,kCAAkC;;YA5HlC,qCAAqC;;YAErC,8BAA8B;;YAE9B,oCAAoC;;YAEpC,kCAAkC;;gBAlDlC,sCAAsC;;gBAEtC,uCAAuC;;gBAEvC,qCAAqC;;;YAgDrC,+BAA+B;;gBAxC/B,qBAAqB;;gBAErB,kCAAkC;;gBAElC,8BAA8B;;gBAE9B,wBAAwB;;;YAoCxB,4CAA4C;;YAE5C,8CAA8C;;YAE9C,sCAAsC;;;QAgHtC,2BAA2B;;YApG3B,8BAA8B;;YAE9B,gCAAgC;;YAEhC,sCAAsC;;;QAkGtC,4CAA4C;;YA1F5C,yBAAyB;;;;;;;;;YASzB,kCAAkC;;gBArBlC,8BAA8B;;gBAE9B,gCAAgC;;gBAEhC,sCAAsC;;;;QAoGtC,yCAAyC;;YAvEzC,wBAAwB;;gBAEpB,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,yBAAyB;;gBAEzB,uBAAuB;;;;;CA2GrB,CAAC"}
+{"version":3,"file":"torch.d.ts","sourceRoot":"","sources":["../../src/torch/torch.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACZ,MAAM,eAAe,CAAC;AAIvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMrD;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,4BAA4B;;IAE5B,yBAAyB;;IAEzB,uBAAuB;;IAEvB,iBAAiB;;EAEnB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,aAAa;IACtB,sCAAsC;;IAEtC,uCAAuC;;IAEvC,qCAAqC;;EAEvC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,kBAAkB;IAC3B,qBAAqB;;IAErB,kCAAkC;;IAElC,8BAA8B;;IAE9B,wBAAwB;;EAE1B,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,kBAAkB;IAC3B,yCAAyC;;IAEzC,yCAAyC;;QA9CzC,4BAA4B;;QAE5B,yBAAyB;;QAEzB,uBAAuB;;QAEvB,iBAAiB;;;IA0CjB,iCAAiC;;IAEjC,mCAAmC;;EAErC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAC7B,qCAAqC;;IAErC,8BAA8B;;IAE9B,oCAAoC;;IAEpC,kCAAkC;;QAlDlC,sCAAsC;;QAEtC,uCAAuC;;QAEvC,qCAAqC;;;IAgDrC,+BAA+B;;QAxC/B,qBAAqB;;QAErB,kCAAkC;;QAElC,8BAA8B;;QAE9B,wBAAwB;;;IAoCxB,4CAA4C;;IAE5C,8CAA8C;;IAE9C,sCAAsC;;EAExC,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAC7B,8BAA8B;;IAE9B,gCAAgC;;IAEhC,sCAAsC;;EAExC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAC7B,yBAAyB;;;;;;;;;IASzB,kCAAkC;;QArBlC,8BAA8B;;QAE9B,gCAAgC;;QAEhC,sCAAsC;;;EAmBxC,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,kBAAkB;IAC3B,wBAAwB;;QAEpB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,yBAAyB;;QAEzB,uBAAuB;;;EAG7B,CAAC;AAMH;;;;;;;;GAQG;AACH,eAAO,MAAM,eAAe;IAnGxB,yCAAyC;;IAEzC,yCAAyC;;QA9CzC,4BAA4B;;QAE5B,yBAAyB;;QAEzB,uBAAuB;;QAEvB,iBAAiB;;;IA0CjB,iCAAiC;;IAEjC,mCAAmC;;;IAQnC,qCAAqC;;IAErC,8BAA8B;;IAE9B,oCAAoC;;IAEpC,kCAAkC;;QAlDlC,sCAAsC;;QAEtC,uCAAuC;;QAEvC,qCAAqC;;;IAgDrC,+BAA+B;;QAxC/B,qBAAqB;;QAErB,kCAAkC;;QAElC,8BAA8B;;QAE9B,wBAAwB;;;IAoCxB,4CAA4C;;IAE5C,8CAA8C;;IAE9C,sCAAsC;;;IAwBtC,yBAAyB;;;;;;;;;IASzB,kCAAkC;;QArBlC,8BAA8B;;QAE9B,gCAAgC;;QAEhC,sCAAsC;;;GA2DzC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,iBAAiB;IAvC1B,wBAAwB;;QAEpB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,yBAAyB;;QAEzB,uBAAuB;;;2DAmC9B,CAAC;AAEF;;;;;;;;;;;;GAYG;AACH,eAAO,MAAM,qBAAqB;IAnI9B,yCAAyC;;IAEzC,yCAAyC;;QA9CzC,4BAA4B;;QAE5B,yBAAyB;;QAEzB,uBAAuB;;QAEvB,iBAAiB;;;IA0CjB,iCAAiC;;IAEjC,mCAAmC;;;IAQnC,qCAAqC;;IAErC,8BAA8B;;IAE9B,oCAAoC;;IAEpC,kCAAkC;;QAlDlC,sCAAsC;;QAEtC,uCAAuC;;QAEvC,qCAAqC;;;IAgDrC,+BAA+B;;QAxC/B,qBAAqB;;QAErB,kCAAkC;;QAElC,8BAA8B;;QAE9B,wBAAwB;;;IAoCxB,4CAA4C;;IAE5C,8CAA8C;;IAE9C,sCAAsC;;;IAwBtC,yBAAyB;;;;;;;;;IASzB,kCAAkC;;QArBlC,8BAA8B;;QAE9B,gCAAgC;;QAEhC,sCAAsC;;;GA2FzC,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,uBAAuB;IAzEhC,wBAAwB;;QAEpB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,yBAAyB;;QAEzB,uBAAuB;;;sEAqE9B,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,eAAO,MAAM,gBAAgB;IAhHzB,wBAAwB;;QAEpB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,yBAAyB;;QAEzB,uBAAuB;;;mFA4G9B,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,eAAO,MAAM,gBAAgB;IArJzB,wBAAwB;;QAEpB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,yBAAyB;;QAEzB,uBAAuB;;;mFAiJ9B,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,UAAU;IACnB,oCAAoC;;IAEpC,uCAAuC;;IAEvC,+BAA+B;;QA5R/B,4BAA4B;;QAE5B,yBAAyB;;QAEzB,uBAAuB;;QAEvB,iBAAiB;;;IAwRjB,yBAAyB;;QAhRzB,sCAAsC;;QAEtC,uCAAuC;;QAEvC,qCAAqC;;;IA8QrC,qBAAqB;;QAtQrB,qBAAqB;;QAErB,kCAAkC;;QAElC,8BAA8B;;QAE9B,wBAAwB;;;IAkQxB,6BAA6B;;QAtP7B,yCAAyC;;QAEzC,yCAAyC;;YA9CzC,4BAA4B;;YAE5B,yBAAyB;;YAEzB,uBAAuB;;YAEvB,iBAAiB;;;QA0CjB,iCAAiC;;QAEjC,mCAAmC;;;IAkPnC,kCAAkC;;QA1OlC,qCAAqC;;QAErC,8BAA8B;;QAE9B,oCAAoC;;QAEpC,kCAAkC;;YAlDlC,sCAAsC;;YAEtC,uCAAuC;;YAEvC,qCAAqC;;;QAgDrC,+BAA+B;;YAxC/B,qBAAqB;;YAErB,kCAAkC;;YAElC,8BAA8B;;YAE9B,wBAAwB;;;QAoCxB,4CAA4C;;QAE5C,8CAA8C;;QAE9C,sCAAsC;;;IA8NtC,2BAA2B;;QAlN3B,8BAA8B;;QAE9B,gCAAgC;;QAEhC,sCAAsC;;;IAgNtC,4CAA4C;;QAxM5C,yBAAyB;;;;;;;;;QASzB,kCAAkC;;YArBlC,8BAA8B;;YAE9B,gCAAgC;;YAEhC,sCAAsC;;;;IAkNtC,yCAAyC;;QArLzC,wBAAwB;;YAEpB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,yBAAyB;;YAEzB,uBAAuB;;;;CA+KrB,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,eAAO,MAAM,KAAK;IACd,sCAAsC;;QApStC,yCAAyC;;QAEzC,yCAAyC;;YA9CzC,4BAA4B;;YAE5B,yBAAyB;;YAEzB,uBAAuB;;YAEvB,iBAAiB;;;QA0CjB,iCAAiC;;QAEjC,mCAAmC;;;QAQnC,qCAAqC;;QAErC,8BAA8B;;QAE9B,oCAAoC;;QAEpC,kCAAkC;;YAlDlC,sCAAsC;;YAEtC,uCAAuC;;YAEvC,qCAAqC;;;QAgDrC,+BAA+B;;YAxC/B,qBAAqB;;YAErB,kCAAkC;;YAElC,8BAA8B;;YAE9B,wBAAwB;;;QAoCxB,4CAA4C;;QAE5C,8CAA8C;;QAE9C,sCAAsC;;;QAwBtC,yBAAyB;;;;;;;;;QASzB,kCAAkC;;YArBlC,8BAA8B;;YAE9B,gCAAgC;;YAEhC,sCAAsC;;;;IA0PtC,gDAAgD;;QA7NhD,wBAAwB;;YAEpB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,yBAAyB;;YAEzB,uBAAuB;;;;IAuN3B,qCAAqC;;QAxSrC,yCAAyC;;QAEzC,yCAAyC;;YA9CzC,4BAA4B;;YAE5B,yBAAyB;;YAEzB,uBAAuB;;YAEvB,iBAAiB;;;QA0CjB,iCAAiC;;QAEjC,mCAAmC;;;QAQnC,qCAAqC;;QAErC,8BAA8B;;QAE9B,oCAAoC;;QAEpC,kCAAkC;;YAlDlC,sCAAsC;;YAEtC,uCAAuC;;YAEvC,qCAAqC;;;QAgDrC,+BAA+B;;YAxC/B,qBAAqB;;YAErB,kCAAkC;;YAElC,8BAA8B;;YAE9B,wBAAwB;;;QAoCxB,4CAA4C;;QAE5C,8CAA8C;;QAE9C,sCAAsC;;;QAwBtC,yBAAyB;;;;;;;;;QASzB,kCAAkC;;YArBlC,8BAA8B;;YAE9B,gCAAgC;;YAEhC,sCAAsC;;;;IA8PtC,+CAA+C;;QAjO/C,wBAAwB;;YAEpB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,yBAAyB;;YAEzB,uBAAuB;;;;IA2N3B,mEAAmE;;QAnOnE,wBAAwB;;YAEpB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,yBAAyB;;YAEzB,uBAAuB;;;;IA6N3B,4DAA4D;;QArO5D,wBAAwB;;YAEpB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,yBAAyB;;YAEzB,uBAAuB;;;;IA+N3B,uBAAuB;;QApEvB,oCAAoC;;QAEpC,uCAAuC;;QAEvC,+BAA+B;;YA5R/B,4BAA4B;;YAE5B,yBAAyB;;YAEzB,uBAAuB;;YAEvB,iBAAiB;;;QAwRjB,yBAAyB;;YAhRzB,sCAAsC;;YAEtC,uCAAuC;;YAEvC,qCAAqC;;;QA8QrC,qBAAqB;;YAtQrB,qBAAqB;;YAErB,kCAAkC;;YAElC,8BAA8B;;YAE9B,wBAAwB;;;QAkQxB,6BAA6B;;YAtP7B,yCAAyC;;YAEzC,yCAAyC;;gBA9CzC,4BAA4B;;gBAE5B,yBAAyB;;gBAEzB,uBAAuB;;gBAEvB,iBAAiB;;;YA0CjB,iCAAiC;;YAEjC,mCAAmC;;;QAkPnC,kCAAkC;;YA1OlC,qCAAqC;;YAErC,8BAA8B;;YAE9B,oCAAoC;;YAEpC,kCAAkC;;gBAlDlC,sCAAsC;;gBAEtC,uCAAuC;;gBAEvC,qCAAqC;;;YAgDrC,+BAA+B;;gBAxC/B,qBAAqB;;gBAErB,kCAAkC;;gBAElC,8BAA8B;;gBAE9B,wBAAwB;;;YAoCxB,4CAA4C;;YAE5C,8CAA8C;;YAE9C,sCAAsC;;;QA8NtC,2BAA2B;;YAlN3B,8BAA8B;;YAE9B,gCAAgC;;YAEhC,sCAAsC;;;QAgNtC,4CAA4C;;YAxM5C,yBAAyB;;;;;;;;;YASzB,kCAAkC;;gBArBlC,8BAA8B;;gBAE9B,gCAAgC;;gBAEhC,sCAAsC;;;;QAkNtC,yCAAyC;;YArLzC,wBAAwB;;gBAEpB,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,yBAAyB;;gBAEzB,uBAAuB;;;;;CAiOrB,CAAC"}

package/dist/torch/torch.js

@@ -161,6 +161,96 @@ export const torch_mlp_train = East.platform("torch_mlp_train", [MatrixType, Vec
  * @returns Predicted values
  */
 export const torch_mlp_predict = East.platform("torch_mlp_predict", [TorchModelBlobType, MatrixType], VectorType);
+/**
+ * Train a PyTorch MLP model with multi-output support.
+ *
+ * Supports multi-output regression (predicting multiple values per sample)
+ * and autoencoders (where input equals target for reconstruction learning).
+ * Output dimension is inferred from y.shape[1] unless overridden in config.
+ *
+ * @param X - Feature matrix (n_samples x n_features)
+ * @param y - Target matrix (n_samples x n_outputs)
+ * @param mlp_config - MLP architecture configuration
+ * @param train_config - Training configuration
+ * @returns Model blob and training result
+ */
+export const torch_mlp_train_multi = East.platform("torch_mlp_train_multi", [MatrixType, MatrixType, TorchMLPConfigType, TorchTrainConfigType], TorchTrainOutputType);
+/**
+ * Make predictions with a trained PyTorch MLP (multi-output).
+ *
+ * Returns a matrix where each row contains the predicted outputs for a sample.
+ *
+ * @param model - Trained MLP model blob
+ * @param X - Feature matrix (n_samples x n_features)
+ * @returns Predicted matrix (n_samples x n_outputs)
+ */
+export const torch_mlp_predict_multi = East.platform("torch_mlp_predict_multi", [TorchModelBlobType, MatrixType], MatrixType);
+/**
+ * Extract intermediate layer activations (embeddings) from a trained MLP.
+ *
+ * For autoencoders, this allows extracting the bottleneck representation.
+ * The layer_index specifies which hidden layer's output to return (0-indexed).
+ *
+ * For an autoencoder with architecture [input -> 8 -> 2 -> 8 -> output]
+ * (hidden_layers: [8, 2, 8]):
+ * - layer_index=0: output after first hidden layer (8 features)
+ * - layer_index=1: output after second hidden layer (2 features) <- bottleneck
+ * - layer_index=2: output after third hidden layer (8 features)
+ *
+ * @param model - Trained MLP model blob
+ * @param X - Feature matrix (n_samples x n_features)
+ * @param layer_index - Which hidden layer's output to return (0-indexed)
+ * @returns Embedding matrix (n_samples x hidden_dim at that layer)
+ *
+ * @example
+ * ```ts
+ * // Train autoencoder: 4 features -> 8 -> 2 (bottleneck) -> 8 -> 4 features
+ * const mlp_config = $.let({
+ *     hidden_layers: [8n, 2n, 8n],
+ *     activation: variant('some', variant('relu', {})),
+ *     dropout: variant('none', null),
+ *     output_dim: variant('none', null),
+ * });
+ * const output = $.let(Torch.mlpTrainMulti(X, X, mlp_config, train_config));
+ *
+ * // Extract bottleneck embeddings (layer_index=1 for the 2-dim bottleneck)
+ * const embeddings = $.let(Torch.mlpEncode(output.model, X, 1n));
+ * // embeddings is now (n_samples x 2)
+ * ```
+ */
+export const torch_mlp_encode = East.platform("torch_mlp_encode", [TorchModelBlobType, MatrixType, IntegerType], MatrixType);
+/**
+ * Decode embeddings back through the decoder portion of an MLP.
+ *
+ * For autoencoders, this takes bottleneck activations and runs them through
+ * the decoder to reconstruct the output. This is the complement to mlpEncode.
+ *
+ * For an autoencoder with architecture [input -> 8 -> 2 -> 8 -> output]
+ * (hidden_layers: [8, 2, 8]):
+ * - layer_index=1: Start from the 2-dim bottleneck, run through layers 2+ to output
+ * - layer_index=0: Start from the 8-dim first layer, run through layers 1+ to output
+ *
+ * Use case: Compute weighted average of origin embeddings, then decode to
+ * get the reconstructed blend weight distribution.
+ *
+ * @param model - Trained MLP model blob
+ * @param embeddings - Embedding matrix (n_samples x hidden_dim at layer_index)
+ * @param layer_index - Which hidden layer the embeddings come from (0-indexed)
+ * @returns Decoded output matrix (n_samples x output_dim)
+ *
+ * @example
+ * ```ts
+ * // After training autoencoder and extracting embeddings...
+ * const origin_embeddings = $.let(Torch.mlpEncode(output.model, X_onehot, 1n));
+ *
+ * // Compute weighted blend embedding (e.g., 50% origin A + 50% origin B)
+ * const blend_embedding = $.let(...); // weighted average of origin embeddings
+ *
+ * // Decode back to weight distribution
+ * const reconstructed = $.let(Torch.mlpDecode(output.model, blend_embedding, 1n));
+ * ```
+ */
+export const torch_mlp_decode = East.platform("torch_mlp_decode", [TorchModelBlobType, MatrixType, IntegerType], MatrixType);
 // ============================================================================
 // Grouped Export
 // ============================================================================
@@ -223,10 +313,18 @@ export const TorchTypes = {
  * ```
  */
 export const Torch = {
-    /** Train MLP model */
+    /** Train MLP model (single output) */
     mlpTrain: torch_mlp_train,
-    /** Make predictions with MLP */
+    /** Make predictions with MLP (single output) */
     mlpPredict: torch_mlp_predict,
+    /** Train MLP model (multi-output) */
+    mlpTrainMulti: torch_mlp_train_multi,
+    /** Make predictions with MLP (multi-output) */
+    mlpPredictMulti: torch_mlp_predict_multi,
+    /** Extract intermediate layer activations (embeddings) from MLP */
+    mlpEncode: torch_mlp_encode,
+    /** Decode embeddings back through decoder portion of MLP */
+    mlpDecode: torch_mlp_decode,
     /** Type definitions */
     Types: TorchTypes,
 };

package/dist/torch/torch.js.map

@@ -1 +1 @@
-{"version":3,"file":"torch.js","sourceRoot":"","sources":["../../src/torch/torch.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,GACZ,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;IAC3C,4BAA4B;IAC5B,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC;IACpB,yBAAyB;IACzB,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC;IACpB,uBAAuB;IACvB,OAAO,EAAE,UAAU,CAAC,EAAE,CAAC;IACvB,iBAAiB;IACjB,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;CAC7B,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,WAAW,CAAC;IACrC,sCAAsC;IACtC,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;IACnB,uCAAuC;IACvC,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;IACnB,qCAAqC;IACrC,aAAa,EAAE,UAAU,CAAC,EAAE,CAAC;CAChC,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;IAC1C,qBAAqB;IACrB,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC;IACpB,kCAAkC;IAClC,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;IACnB,8BAA8B;IAC9B,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC;IACrB,wBAAwB;IACxB,OAAO,EAAE,UAAU,CAAC,EAAE,CAAC;CAC1B,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,UAAU,CAAC;IACzC,yCAAyC;IACzC,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;IACrC,yCAAyC;IACzC,UAAU,EAAE,UAAU,CAAC,mBAAmB,CAAC;IAC3C,iCAAiC;IACjC,OAAO,EAAE,UAAU,CAAC,SAAS,CAAC;IAC9B,mCAAmC;IACnC,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;CACtC,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,UAAU,CAAC;IAC3C,qCAAqC;IACrC,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;IAC/B,8BAA8B;IAC9B,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;IACnC,oCAAoC;IACpC,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,kCAAkC;IAClC,IAAI,EAAE,UAAU,CAAC,aAAa,CAAC;IAC/B,+BAA+B;IAC/B,SAAS,EAAE,UAAU,CAAC,kBAAkB,CAAC;IACzC,4CAA4C;IAC5C,cAAc,EAAE,UAAU,CAAC,WAAW,CAAC;IACvC,8CAA8C;IAC9C,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;IACvC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;CACxC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,UAAU,CAAC;IAC3C,8BAA8B;IAC9B,YAAY,EAAE,UAAU;IACxB,gCAAgC;IAChC,UAAU,EAAE,UAAU;IACtB,sCAAsC;IACtC,UAAU,EAAE,WAAW;CAC1B,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,UAAU,CAAC;IAC3C,yBAAyB;IACzB,KAAK,EAAE,WAAW,CAAC;QACf,SAAS,EAAE,UAAU,CAAC;YAClB,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,WAAW;YACvB,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;YACrC,UAAU,EAAE,WAAW;SAC1B,CAAC;KACL,CAAC;IACF,kCAAkC;IAClC,MAAM,EAAE,oBAAoB;CAC/B,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;IAC1C,wBAAwB;IACxB,SAAS,EAAE,UAAU,CAAC;QAClB,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,yBAAyB;QACzB,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;QACrC,uBAAuB;QACvB,UAAU,EAAE,WAAW;KAC1B,CAAC;CACL,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB,CAAC,UAAU,EAAE,UAAU,EAAE,kBAAkB,EAAE,oBAAoB,CAAC,EAClE,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAC1C,mBAAmB,EACnB,CAAC,kBAAkB,EAAE,UAAU,CAAC,EAChC,UAAU,CACb,CAAC;AAEF,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG;IACtB,oCAAoC;IACpC,UAAU;IACV,uCAAuC;IACvC,UAAU;IACV,+BAA+B;IAC/B,mBAAmB;IACnB,yBAAyB;IACzB,aAAa;IACb,qBAAqB;IACrB,kBAAkB;IAClB,6BAA6B;IAC7B,kBAAkB;IAClB,kCAAkC;IAClC,oBAAoB;IACpB,2BAA2B;IAC3B,oBAAoB;IACpB,4CAA4C;IAC5C,oBAAoB;IACpB,yCAAyC;IACzC,aAAa,EAAE,kBAAkB;CAC3B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,CAAC,MAAM,KAAK,GAAG;IACjB,sBAAsB;IACtB,QAAQ,EAAE,eAAe;IACzB,gCAAgC;IAChC,UAAU,EAAE,iBAAiB;IAC7B,uBAAuB;IACvB,KAAK,EAAE,UAAU;CACX,CAAC"}
+{"version":3,"file":"torch.js","sourceRoot":"","sources":["../../src/torch/torch.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,GACZ,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;IAC3C,4BAA4B;IAC5B,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC;IACpB,yBAAyB;IACzB,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC;IACpB,uBAAuB;IACvB,OAAO,EAAE,UAAU,CAAC,EAAE,CAAC;IACvB,iBAAiB;IACjB,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;CAC7B,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG,WAAW,CAAC;IACrC,sCAAsC;IACtC,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;IACnB,uCAAuC;IACvC,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;IACnB,qCAAqC;IACrC,aAAa,EAAE,UAAU,CAAC,EAAE,CAAC;CAChC,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;IAC1C,qBAAqB;IACrB,IAAI,EAAE,UAAU,CAAC,EAAE,CAAC;IACpB,kCAAkC;IAClC,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;IACnB,8BAA8B;IAC9B,KAAK,EAAE,UAAU,CAAC,EAAE,CAAC;IACrB,wBAAwB;IACxB,OAAO,EAAE,UAAU,CAAC,EAAE,CAAC;CAC1B,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,UAAU,CAAC;IACzC,yCAAyC;IACzC,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;IACrC,yCAAyC;IACzC,UAAU,EAAE,UAAU,CAAC,mBAAmB,CAAC;IAC3C,iCAAiC;IACjC,OAAO,EAAE,UAAU,CAAC,SAAS,CAAC;IAC9B,mCAAmC;IACnC,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;CACtC,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,UAAU,CAAC;IAC3C,qCAAqC;IACrC,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;IAC/B,8BAA8B;IAC9B,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;IACnC,oCAAoC;IACpC,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,kCAAkC;IAClC,IAAI,EAAE,UAAU,CAAC,aAAa,CAAC;IAC/B,+BAA+B;IAC/B,SAAS,EAAE,UAAU,CAAC,kBAAkB,CAAC;IACzC,4CAA4C;IAC5C,cAAc,EAAE,UAAU,CAAC,WAAW,CAAC;IACvC,8CAA8C;IAC9C,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;IACvC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;CACxC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,UAAU,CAAC;IAC3C,8BAA8B;IAC9B,YAAY,EAAE,UAAU;IACxB,gCAAgC;IAChC,UAAU,EAAE,UAAU;IACtB,sCAAsC;IACtC,UAAU,EAAE,WAAW;CAC1B,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,UAAU,CAAC;IAC3C,yBAAyB;IACzB,KAAK,EAAE,WAAW,CAAC;QACf,SAAS,EAAE,UAAU,CAAC;YAClB,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE,WAAW;YACvB,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;YACrC,UAAU,EAAE,WAAW;SAC1B,CAAC;KACL,CAAC;IACF,kCAAkC;IAClC,MAAM,EAAE,oBAAoB;CAC/B,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,WAAW,CAAC;IAC1C,wBAAwB;IACxB,SAAS,EAAE,UAAU,CAAC;QAClB,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,yBAAyB;QACzB,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;QACrC,uBAAuB;QACvB,UAAU,EAAE,WAAW;KAC1B,CAAC;CACL,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB,CAAC,UAAU,EAAE,UAAU,EAAE,kBAAkB,EAAE,oBAAoB,CAAC,EAClE,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAC1C,mBAAmB,EACnB,CAAC,kBAAkB,EAAE,UAAU,CAAC,EAChC,UAAU,CACb,CAAC;AAEF;;;;;;;;;;;;GAYG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAC9C,uBAAuB,EACvB,CAAC,UAAU,EAAE,UAAU,EAAE,kBAAkB,EAAE,oBAAoB,CAAC,EAClE,oBAAoB,CACvB,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,IAAI,CAAC,QAAQ,CAChD,yBAAyB,EACzB,CAAC,kBAAkB,EAAE,UAAU,CAAC,EAChC,UAAU,CACb,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CACzC,kBAAkB,EAClB,CAAC,kBAAkB,EAAE,UAAU,EAAE,WAAW,CAAC,EAC7C,UAAU,CACb,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8BG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CACzC,kBAAkB,EAClB,CAAC,kBAAkB,EAAE,UAAU,EAAE,WAAW,CAAC,EAC7C,UAAU,CACb,CAAC;AAEF,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG;IACtB,oCAAoC;IACpC,UAAU;IACV,uCAAuC;IACvC,UAAU;IACV,+BAA+B;IAC/B,mBAAmB;IACnB,yBAAyB;IACzB,aAAa;IACb,qBAAqB;IACrB,kBAAkB;IAClB,6BAA6B;IAC7B,kBAAkB;IAClB,kCAAkC;IAClC,oBAAoB;IACpB,2BAA2B;IAC3B,oBAAoB;IACpB,4CAA4C;IAC5C,oBAAoB;IACpB,yCAAyC;IACzC,aAAa,EAAE,kBAAkB;CAC3B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAgCG;AACH,MAAM,CAAC,MAAM,KAAK,GAAG;IACjB,sCAAsC;IACtC,QAAQ,EAAE,eAAe;IACzB,gDAAgD;IAChD,UAAU,EAAE,iBAAiB;IAC7B,qCAAqC;IACrC,aAAa,EAAE,qBAAqB;IACpC,+CAA+C;IAC/C,eAAe,EAAE,uBAAuB;IACxC,mEAAmE;IACnE,SAAS,EAAE,gBAAgB;IAC3B,4DAA4D;IAC5D,SAAS,EAAE,gBAAgB;IAC3B,uBAAuB;IACvB,KAAK,EAAE,UAAU;CACX,CAAC"}