Update constants.ts

constants.ts

@@ -11,7 +11,7 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
     category: 'Core',
     parameters: [
       { name: 'name', type: 'string', label: 'Name', default: 'input_1', description: 'Variable name in forward()' },
-      { name: 'modality', type: 'select', label: 'Modality', default: 'Tensor', options: ['Tensor', 'Image', 'Text', 'Audio', 'Video', 'Latent', 'State', '3D Volume', 'Point Cloud', 'Radar', 'Lidar'] },
+      { name: 'modality', type: 'select', label: 'Modality', default: 'Tensor', options: ['Tensor', 'Image', 'Text', 'Audio', 'Video', 'Latent', 'State', '3D Volume', 'Point Cloud', 'Radar', 'Lidar', 'Graph', 'Molecule'] },
       { name: 'shape', type: 'string', label: 'Shape (e.g. 3,224,224)', default: '3, 224, 224' },
       { name: 'batch_size', type: 'number', label: 'Batch Size', default: 32 },
       { name: 'dtype', type: 'select', label: 'Data Type', default: 'float32', options: ['float32', 'int64', 'bool'] }
@@ -62,6 +62,249 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
     ]
   },
 
+  // --- GRAPH NEURAL NETWORKS ---
+  [LayerType.GCN_CONV]: {
+    type: LayerType.GCN_CONV,
+    label: 'GCN Conv',
+    description: 'Graph Convolutional Network',
+    category: 'Graph',
+    parameters: [
+      { name: 'in_channels', type: 'number', label: 'In Channels', default: 16 },
+      { name: 'out_channels', type: 'number', label: 'Out Channels', default: 32 },
+      { name: 'improved', type: 'boolean', label: 'Improved GCN', default: false }
+    ]
+  },
+  [LayerType.GRAPH_SAGE]: {
+    type: LayerType.GRAPH_SAGE,
+    label: 'GraphSAGE',
+    description: 'Inductive Graph Learning',
+    category: 'Graph',
+    parameters: [
+      { name: 'in_channels', type: 'number', label: 'In Channels', default: 16 },
+      { name: 'out_channels', type: 'number', label: 'Out Channels', default: 32 },
+      { name: 'aggr', type: 'select', label: 'Aggregator', default: 'mean', options: ['mean', 'max', 'lstm'] }
+    ]
+  },
+  [LayerType.GAT_CONV]: {
+    type: LayerType.GAT_CONV,
+    label: 'GAT Conv',
+    description: 'Graph Attention Network',
+    category: 'Graph',
+    parameters: [
+      { name: 'in_channels', type: 'number', label: 'In Channels', default: 16 },
+      { name: 'out_channels', type: 'number', label: 'Out Channels', default: 8 },
+      { name: 'heads', type: 'number', label: 'Attention Heads', default: 4 },
+      { name: 'concat', type: 'boolean', label: 'Concat Heads', default: true }
+    ]
+  },
+  [LayerType.GIN_CONV]: {
+    type: LayerType.GIN_CONV,
+    label: 'GIN Conv',
+    description: 'Graph Isomorphism Network',
+    category: 'Graph',
+    parameters: [
+      { name: 'eps', type: 'number', label: 'Epsilon', default: 0 },
+      { name: 'train_eps', type: 'boolean', label: 'Train Epsilon', default: true }
+    ]
+  },
+
+  // --- PHYSICS & SCIML ---
+  [LayerType.NEURAL_ODE]: {
+    type: LayerType.NEURAL_ODE,
+    label: 'Neural ODE',
+    description: 'Continuous Depth Model',
+    category: 'Physics',
+    parameters: [
+      { name: 'hidden_dim', type: 'number', label: 'Hidden Dim', default: 64 },
+      { name: 'solver', type: 'select', label: 'Solver', default: 'dopri5', options: ['dopri5', 'rk4', 'euler'] }
+    ]
+  },
+  [LayerType.PINN_LINEAR]: {
+    type: LayerType.PINN_LINEAR,
+    label: 'PINN Linear',
+    description: 'Physics-Informed Linear Layer',
+    category: 'Physics',
+    parameters: [
+      { name: 'in_features', type: 'number', label: 'In Features', default: 32 },
+      { name: 'out_features', type: 'number', label: 'Out Features', default: 32 },
+      { name: 'constraint', type: 'text', label: 'Constraint Formula', default: 'grad(u, x) - f(x) = 0' }
+    ]
+  },
+  [LayerType.HAMILTONIAN_NN]: {
+    type: LayerType.HAMILTONIAN_NN,
+    label: 'Hamiltonian NN',
+    description: 'Conserves Energy/Momentum',
+    category: 'Physics',
+    parameters: [
+      { name: 'dim', type: 'number', label: 'Dimension', default: 2 },
+      { name: 'hidden_dim', type: 'number', label: 'Hidden Dim', default: 64 }
+    ]
+  },
+  [LayerType.PROTEIN_FOLDING]: {
+    type: LayerType.PROTEIN_FOLDING,
+    label: 'Protein Folding',
+    description: 'AlphaFold EvoFormer Block',
+    category: 'Physics',
+    parameters: [
+      { name: 'msa_dim', type: 'number', label: 'MSA Dim', default: 256 },
+      { name: 'pair_dim', type: 'number', label: 'Pair Dim', default: 128 },
+      { name: 'num_heads', type: 'number', label: 'Heads', default: 8 }
+    ]
+  },
+
+  // --- SPIKING & NEUROMORPHIC ---
+  [LayerType.LIF_NEURON]: {
+    type: LayerType.LIF_NEURON,
+    label: 'LIF Neuron',
+    description: 'Leaky Integrate-and-Fire',
+    category: 'Spiking',
+    parameters: [
+      { name: 'tau', type: 'number', label: 'Time Constant', default: 2.0 },
+      { name: 'v_threshold', type: 'number', label: 'Threshold', default: 1.0 },
+      { name: 'surrogate_grad', type: 'select', label: 'Surrogate Grad', default: 'fast_sigmoid', options: ['fast_sigmoid', 'arctan'] }
+    ]
+  },
+  [LayerType.SPIKING_LAYER]: {
+    type: LayerType.SPIKING_LAYER,
+    label: 'Spiking Dense',
+    description: 'Fully Connected SNN Layer',
+    category: 'Spiking',
+    parameters: [
+      { name: 'in_features', type: 'number', label: 'In Features', default: 128 },
+      { name: 'out_features', type: 'number', label: 'Out Features', default: 128 },
+      { name: 'spike_mode', type: 'boolean', label: 'Output Spikes', default: true }
+    ]
+  },
+
+  // --- REINFORCEMENT LEARNING ---
+  [LayerType.DUELING_HEAD]: {
+    type: LayerType.DUELING_HEAD,
+    label: 'Dueling Head',
+    description: 'Separates Value and Advantage',
+    category: 'RL',
+    parameters: [
+      { name: 'input_dim', type: 'number', label: 'Input Dim', default: 512 },
+      { name: 'action_dim', type: 'number', label: 'Action Dim', default: 4 }
+    ]
+  },
+  [LayerType.PPO_HEAD]: {
+    type: LayerType.PPO_HEAD,
+    label: 'PPO Actor-Critic',
+    description: 'Policy and Value Heads',
+    category: 'RL',
+    parameters: [
+      { name: 'hidden_dim', type: 'number', label: 'Hidden Dim', default: 256 },
+      { name: 'action_space', type: 'select', label: 'Action Space', default: 'Discrete', options: ['Discrete', 'Continuous'] }
+    ]
+  },
+  [LayerType.SAC_HEAD]: {
+    type: LayerType.SAC_HEAD,
+    label: 'SAC Head',
+    description: 'Soft Actor-Critic Output',
+    category: 'RL',
+    parameters: [
+      { name: 'hidden_dim', type: 'number', label: 'Hidden Dim', default: 256 },
+      { name: 'log_std_min', type: 'number', label: 'Min Log Std', default: -20 },
+      { name: 'log_std_max', type: 'number', label: 'Max Log Std', default: 2 }
+    ]
+  },
+
+  // --- ADVANCED / NICHE ---
+  [LayerType.CAPSULE]: {
+    type: LayerType.CAPSULE,
+    label: 'Capsule Layer',
+    description: 'Preserves spatial hierarchy',
+    category: 'Advanced',
+    parameters: [
+      { name: 'num_capsules', type: 'number', label: 'Num Capsules', default: 10 },
+      { name: 'capsule_dim', type: 'number', label: 'Capsule Dim', default: 16 },
+      { name: 'routings', type: 'number', label: 'Routing Iters', default: 3 }
+    ]
+  },
+  [LayerType.HYPER_NET]: {
+    type: LayerType.HYPER_NET,
+    label: 'HyperNetwork',
+    description: 'Generates weights for another NN',
+    category: 'Advanced',
+    parameters: [
+      { name: 'target_shape', type: 'string', label: 'Target Shape', default: '64,64' },
+      { name: 'embedding_dim', type: 'number', label: 'Z Dim', default: 32 }
+    ]
+  },
+  [LayerType.MAMBA_BLOCK]: {
+    type: LayerType.MAMBA_BLOCK,
+    label: 'Mamba (SSM)',
+    description: 'Selective State Space Model',
+    category: 'Advanced',
+    parameters: [
+      { name: 'd_model', type: 'number', label: 'D Model', default: 512 },
+      { name: 'd_state', type: 'number', label: 'State Dim', default: 16 },
+      { name: 'expand', type: 'number', label: 'Expansion', default: 2 }
+    ]
+  },
+  [LayerType.RWKV_BLOCK]: {
+    type: LayerType.RWKV_BLOCK,
+    label: 'RWKV Block',
+    description: 'RNN-Transformer Hybrid',
+    category: 'Advanced',
+    parameters: [
+      { name: 'n_embd', type: 'number', label: 'Embed Dim', default: 768 },
+      { name: 'n_layer', type: 'number', label: 'Layer Idx', default: 0 }
+    ]
+  },
+  [LayerType.HOPFIELD]: {
+    type: LayerType.HOPFIELD,
+    label: 'Hopfield Layer',
+    description: 'Dense Associative Memory',
+    category: 'Advanced',
+    parameters: [
+      { name: 'in_features', type: 'number', label: 'Features', default: 64 },
+      { name: 'stored_patterns', type: 'number', label: 'Pattern Capacity', default: 10 }
+    ]
+  },
+  [LayerType.NORMALIZING_FLOW]: {
+    type: LayerType.NORMALIZING_FLOW,
+    label: 'Normalizing Flow',
+    description: 'Invertible Generative Model',
+    category: 'Advanced',
+    parameters: [
+      { name: 'flow_type', type: 'select', label: 'Type', default: 'RealNVP', options: ['RealNVP', 'Glow', 'MAF'] },
+      { name: 'num_flows', type: 'number', label: 'Num Flows', default: 4 }
+    ]
+  },
+  [LayerType.DNC_MEMORY]: {
+    type: LayerType.DNC_MEMORY,
+    label: 'DNC Memory',
+    description: 'Differentiable Neural Computer',
+    category: 'Advanced',
+    parameters: [
+      { name: 'memory_size', type: 'number', label: 'Mem Slots', default: 128 },
+      { name: 'word_size', type: 'number', label: 'Word Size', default: 20 },
+      { name: 'num_read_heads', type: 'number', label: 'Read Heads', default: 4 }
+    ]
+  },
+  [LayerType.ARCFACE]: {
+    type: LayerType.ARCFACE,
+    label: 'ArcFace',
+    description: 'Metric Learning Loss',
+    category: 'Advanced',
+    parameters: [
+      { name: 's', type: 'number', label: 'Scale (s)', default: 64.0 },
+      { name: 'm', type: 'number', label: 'Margin (m)', default: 0.5 },
+      { name: 'num_classes', type: 'number', label: 'Classes', default: 1000 }
+    ]
+  },
+  [LayerType.ECHO_STATE]: {
+    type: LayerType.ECHO_STATE,
+    label: 'Reservoir (ESN)',
+    description: 'Echo State Network',
+    category: 'Advanced',
+    parameters: [
+      { name: 'reservoir_size', type: 'number', label: 'Reservoir Size', default: 1000 },
+      { name: 'spectral_radius', type: 'number', label: 'Spectral Radius', default: 0.9 }
+    ]
+  },
+
   // --- VIDEO / GENERATION ---
   [LayerType.VIDEO_DIFFUSION_BLOCK]: {
     type: LayerType.VIDEO_DIFFUSION_BLOCK,
@@ -410,6 +653,17 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
       { name: 'embedding_dim', type: 'number', label: 'Embedding Dim', default: 512 }
     ]
   },
+  [LayerType.SINC_CONV]: {
+    type: LayerType.SINC_CONV,
+    label: 'SincConv',
+    description: 'Parametric Sinc Filters (Speech)',
+    category: 'Audio',
+    parameters: [
+      { name: 'out_channels', type: 'number', label: 'Filters', default: 80 },
+      { name: 'kernel_size', type: 'number', label: 'Kernel Size', default: 251 },
+      { name: 'min_low_hz', type: 'number', label: 'Min Hz', default: 50 }
+    ]
+  },
 
   // --- 3D / VISION ---
   [LayerType.NERF_BLOCK]: {
@@ -463,6 +717,16 @@ export const LAYER_DEFINITIONS: Record<LayerType, LayerDefinition> = {
       { name: 'sh_degree', type: 'number', label: 'SH Degree', default: 3 }
     ]
   },
+  [LayerType.MESH_CONV]: {
+    type: LayerType.MESH_CONV,
+    label: 'Mesh Conv',
+    description: 'Convolution on 3D Meshes',
+    category: '3D',
+    parameters: [
+      { name: 'in_channels', type: 'number', label: 'In Channels', default: 3 },
+      { name: 'out_channels', type: 'number', label: 'Out Channels', default: 64 }
+    ]
+  },
 
   // --- MERGE ---
   [LayerType.CONCAT]: {
@@ -1311,4 +1575,4 @@ export const TEMPLATES: Record<string, GraphTemplate> = {
       { id: '4', source: 'cat', target: 'unet' }, { id: '5', source: 'unet', target: 'out' }
     ]
   }
-};
+};