Add cortex/__init__.py

cortex/__init__.py

@@ -0,0 +1,32 @@
+"""
+Cortex: Modular Cognitive Plug-ins for Pretrained LLMs
+======================================================
+
+A framework for surgically inserting new cognitive capabilities into 
+pretrained transformer language models via layer surgery.
+
+Modules:
+    1. MemoryBank       — Persistent episodic memory via gated cross-attention (LM2-inspired)
+    2. HallucinationGate — Per-token confidence estimation & gated suppression (InternalInspector-inspired)
+    3. PauseAndThink    — Learnable latent "thinking" tokens for extra compute (Pause Token paper)
+    4. BacktrackHead    — Learned backtracking via early-exit confidence routing (GateSkip + River-LLM)
+    5. SteeringVector   — Activation-space behavioral control (Representation Engineering)
+    6. AdaptiveDepth    — Dynamic layer skipping with learned gates (Mixture of Depths)
+
+All modules follow the CortexModule protocol and can be composed freely.
+"""
+
+from cortex.core import CortexModule, CortexSurgeon, CortexConfig
+from cortex.memory_bank import MemoryBank
+from cortex.hallucination_gate import HallucinationGate
+from cortex.pause_and_think import PauseAndThink
+from cortex.backtrack_head import BacktrackHead
+from cortex.steering_vector import SteeringVector
+from cortex.adaptive_depth import AdaptiveDepth
+
+__version__ = "0.1.0"
+__all__ = [
+    "CortexModule", "CortexSurgeon", "CortexConfig",
+    "MemoryBank", "HallucinationGate", "PauseAndThink",
+    "BacktrackHead", "SteeringVector", "AdaptiveDepth",
+]