From c90b48e3f8da9dd0f8d2ae82ddf977436bb0cfc3 Mon Sep 17 00:00:00 2001 From: YurenHao0426 Date: Sun, 15 Feb 2026 18:19:50 +0000 Subject: Initial implementation of HAG (Hopfield-Augmented Generation) Core Hopfield retrieval module with energy-based convergence guarantees, memory bank, FAISS baseline retriever, evaluation metrics, and end-to-end pipeline. All 45 tests passing on CPU with synthetic data. Co-Authored-By: Claude Opus 4.6 --- hag/__init__.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 hag/__init__.py (limited to 'hag/__init__.py') diff --git a/hag/__init__.py b/hag/__init__.py new file mode 100644 index 0000000..18496e9 --- /dev/null +++ b/hag/__init__.py @@ -0,0 +1,21 @@ +"""HAG: Hopfield-Augmented Generation.""" + +from hag.config import ( + EncoderConfig, + GeneratorConfig, + HopfieldConfig, + MemoryBankConfig, + PipelineConfig, +) +from hag.datatypes import HopfieldResult, PipelineResult, RetrievalResult + +__all__ = [ + "HopfieldConfig", + "MemoryBankConfig", + "EncoderConfig", + "GeneratorConfig", + "PipelineConfig", + "HopfieldResult", + "RetrievalResult", + "PipelineResult", +] -- cgit v1.2.3