kolmgformers 0.0.3__tar.gz

kolmgformers-0.0.3/LICENSE

@@ -0,0 +1,179 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship made available under
+      the License, as indicated by a copyright notice that is included in
+      or attached to the work (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean, as submitted to the Licensor for inclusion
+      in the Work by the copyright owner or by an individual or Legal Entity
+      authorized to submit on behalf of the copyright owner. For the purposes
+      of this definition, "submitted" means any form of electronic, verbal,
+      or written communication sent to the Licensor or its representatives,
+      including but not limited to communication on electronic mailing lists,
+      source code control systems, and issue tracking systems that are managed
+      by, or on behalf of, the Licensor for the purpose of discussing and
+      improving the Work, but excluding communication that is conspicuously
+      marked or designated in writing by the copyright owner as "Not a
+      Contribution."
+
+      "Contributor" shall mean Licensor and any Legal Entity on behalf of
+      whom a Contribution has been received by the Licensor and included
+      within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by the combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a cross-claim
+      or counterclaim in a lawsuit) alleging that the Work or any
+      Contribution embodied within the Work constitutes direct or
+      contributory patent infringement, then any patent licenses granted to
+      You under this License for that Work shall terminate as of the date
+      such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or Derivative Works
+          a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, You must include a readable copy of the attribution
+          notices contained within such NOTICE file, in at least one of the
+          following places: within a NOTICE text provided as part of the
+          Derivative Works; within the Source form or documentation, if
+          provided along with the Derivative Works; or, within a display
+          generated by the Derivative Works, if and wherever such
+          third-party notices normally appear. The contents of the NOTICE
+          file are for informational purposes only and do not modify the
+          License. You may add Your own attribution notices within
+          Derivative Works that You distribute, alongside or as an addendum
+          to the NOTICE text from the Work, provided that such additional
+          attribution notices cannot be construed as modifying the License.
+
+      You may add Your own license statement for Your modifications and
+      may provide additional grant of rights to use, copy, modify, merge,
+      publish, distribute, sublicense, and/or sell copies of the
+      Contribution, subject to the terms and conditions of this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or agreed
+      to in writing, Licensor provides the Work (and each Contributor
+      provides its Contributions) on an "AS IS" BASIS, WITHOUT WARRANTIES
+      OR CONDITIONS OF ANY KIND, either express or implied, including,
+      without limitation, any warranties or conditions of TITLE,
+      NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A PARTICULAR
+      PURPOSE. You are solely responsible for determining the
+      appropriateness of using or reproducing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or exemplary damages of any character arising as a result
+      of this License or out of the use or inability to use the Work
+      (including but not limited to damages for loss of goodwill, work
+      stoppage, computer failure or malfunction, or all other commercial
+      damages or losses), even if such Contributor has been advised of the
+      possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   Copyright 2024 OMGFormers Contributors
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

kolmgformers-0.0.3/MANIFEST.in

@@ -0,0 +1,4 @@
+include README.md
+include LICENSE
+include pyproject.toml
+recursive-include kolmgformers *.py *.typed

kolmgformers-0.0.3/PKG-INFO

@@ -0,0 +1,219 @@
+Metadata-Version: 2.4
+Name: kolmgformers
+Version: 0.0.3
+Summary: KOLMGformers: Unified KAN attention-free sequence modeling (KOLMOGformers) + Parallel Diffusion LM (OMGformers)
+Author: Ömür Bera Işık
+License-Expression: Apache-2.0
+Keywords: deep-learning,transformers,diffusion,language-model,nlp,pytorch,kolmogorov-arnold,kan,attention-free,lora,moe
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.0
+Provides-Extra: hf
+Requires-Dist: transformers>=4.35; extra == "hf"
+Requires-Dist: tokenizers>=0.15; extra == "hf"
+Provides-Extra: safetensors
+Requires-Dist: safetensors>=0.4; extra == "safetensors"
+Provides-Extra: flash
+Requires-Dist: flash-attn>=2.0; extra == "flash"
+Provides-Extra: quant
+Requires-Dist: bitsandbytes>=0.41; extra == "quant"
+Provides-Extra: all
+Requires-Dist: transformers>=4.35; extra == "all"
+Requires-Dist: tokenizers>=0.15; extra == "all"
+Requires-Dist: safetensors>=0.4; extra == "all"
+Requires-Dist: bitsandbytes>=0.41; extra == "all"
+Dynamic: license-file
+
+# KOLMGformers  v0.0.3
+
+Unified Python library merging two research-grade model families:
+
+| Family | Architecture | Key property |
+|--------|-------------|-------------|
+| **KOLMOG** | Kolmogorov-Arnold Cumulative Context | Attention-free, O(d) memory |
+| **OMG** | Parallel Diffusion Transformer | Masked diffusion, full feature set |
+
+---
+
+## What's New in v0.0.3 — Bug Fixes & Improvements
+
+### Bug Fixes (KOLMOG)
+
+| ID | Component | Issue | Fix |
+|----|-----------|-------|-----|
+| #K5 | `KOLMOGformerLayer` | `phi` received raw `kappa_out` and `context` — shape contract was implicit and fragile | Documented and type-checked; context extractor now propagates `attention_mask` |
+| #K6 | `generate()` | Repetition penalty used a Python loop over `set(generated[b].tolist())` — O(seq·vocab) per step | Vectorised with `tensor.unique()` + `scatter_` |
+| #K7 | `generate()` | Top-p nucleus sampling: `cumprobs − softmax(sorted)` double-subtracted the pivot token, causing off-by-one exclusions | Replaced with correct shifted-cumsum implementation |
+| #K8 | `PLKANLayer` | `breakpoints` initialized via `expand().clone()` left non-contiguous memory; subtle autograd issues under in-place ops | Replaced with `linspace(...).repeat()` → always contiguous |
+| #K9 | `InnerKolmogorovFunction` | Always used slow B-spline `KANLayer` for φ layers, ignoring `config.use_plkan` | `build_kan_layer` factory now honoured for φ too (~3–5× speedup with PLKAN) |
+| #K10 | `CumulativeContextExtractor` | Causal pad+shift produced wrong exclusive prefix at position 0 (current token leaked into its own context) | Replaced with correct exclusive cumsum: `C^{<i} = cumsum[i] − kappa_w[i]` |
+| #K11 | `CumulativeContextExtractor` | `attention_mask` was accepted by model but never threaded to the context extractor; pad tokens polluted context vectors | Mask is now applied to `kappa` before accumulation at every layer |
+| #K12 | `KOLMOGformerForCausalLM` | Logits returned only `[:, :-1, :]` (already shifted), breaking downstream use of the full logit tensor | Full-sequence logits returned; shift applied only inside loss computation (matches HF API) |
+| #K13 | `save_pretrained` | Direct `torch.save` to final path could leave a corrupted checkpoint on interruption | Atomic write via `tempfile` + `os.replace`; safetensors support added |
+| #K14 | `KOLMOGformerModel` | No gradient checkpointing — OOM on long sequences during training | `enable_gradient_checkpointing()` added; controlled via `TrainingArguments.gradient_checkpointing` |
+| #K15 | `KANLayer.b_splines` | Grid buffer could be float32 while activations are bfloat16/float16, causing dtype mismatch | Grid is cast to `x.dtype` on every forward pass |
+
+### Bug Fixes (Training)
+
+| ID | Component | Issue | Fix |
+|----|-----------|-------|-----|
+| #T1 | `Trainer` | No early stopping — training continued even after convergence | `early_stopping_patience` added to `TrainingArguments` |
+| #T2 | `DataCollatorForCausalLM` | Sequences silently truncated or accepted without warning | `warn_length` parameter warns when batch sequences exceed model's position limit |
+| #T3 | `Trainer._save_checkpoint` | Saved only model weights — optimizer/scheduler lost, training couldn't truly resume | Optimizer + scheduler state saved in `trainer_state.pt` |
+| #T4 | `Trainer.load_checkpoint` | Restored only model weights and step count | Now restores optimizer, scheduler, early-stopping state |
+| #T5 | `get_scheduler` | Missing `"constant_with_warmup"` type | Added |
+| #T6 | `Trainer` | `bf16=True` on CPU silently fell back to float32 with no warning | Warning emitted; autocast errors caught gracefully |
+| #T7 | `DataCollatorForMaskedLM` | Random-replacement tokens were drawn from `[0, vocab_size)` including `[PAD]`/`[BOS]`/`[EOS]` | Now draws from `[num_special_tokens, vocab_size)` |
+
+### New Features (v0.0.3)
+
+**`KOLMOGformerConfig` additions:**
+- `context_dropout` — independent dropout on the context vector path (default `0.0`).
+- `ffn_type` — FFN activation: `"gelu"` (default) | `"silu"` | `"swiglu"`.
+- `max_position_embeddings_dynamic` — RoPE cache auto-extends beyond limit instead of erroring (default `True`).
+- `validate()` — called in `__post_init__`; surfaces config errors early with helpful messages.
+- `__repr__` — readable summary of key config fields.
+
+**`TrainingArguments` additions:**
+- `early_stopping_patience` — stop after N evaluations without improvement.
+- `gradient_checkpointing` — enable memory-efficient training automatically.
+
+---
+
+## Installation
+
+```bash
+pip install -e .
+# Optional extras
+pip install -e ".[hf]"       # HuggingFace tokenizers
+pip install -e ".[flash]"    # Flash Attention 2
+pip install -e ".[all]"      # Everything
+```
+
+---
+
+## Quick Start
+
+### KOLMOG — Attention-Free Causal LM
+
+```python
+from kolmgformers import KOLMOGformerConfig, KOLMOGformerForCausalLM
+import torch
+
+config = KOLMOGformerConfig(
+    vocab_size=32000,
+    hidden_size=512,
+    num_channels=8,
+    num_layers=6,
+    causal=True,
+    use_nce=True,    # Normalized Context Extraction
+    use_wcc=True,    # Weighted Cumulative Context (v0.0.2+)
+    use_plkan=True,  # Piecewise Linear KAN — 3-5x faster (v0.0.2+)
+)
+model = KOLMOGformerForCausalLM(config)
+print(config)  # v0.0.3: readable repr
+
+ids = torch.tensor([[1, 42, 100]])
+out = model.generate(ids, max_new_tokens=50, temperature=0.8)
+```
+
+### KOLMOG — Training with Early Stopping
+
+```python
+from kolmgformers import (
+    KOLMOGTrainer, KOLMOGTrainingArguments,
+    KOLMOGDataCollatorForCausalLM,
+)
+
+args = KOLMOGTrainingArguments(
+    output_dir="runs/my_run",
+    num_train_epochs=10,
+    early_stopping_patience=3,   # v0.0.3: stop after 3 bad evals
+    gradient_checkpointing=True, # v0.0.3: save memory on long seqs
+    evaluation_strategy="steps",
+    eval_steps=500,
+)
+trainer = KOLMOGTrainer(
+    model=model,
+    args=args,
+    train_dataset=train_ds,
+    eval_dataset=val_ds,
+    data_collator=KOLMOGDataCollatorForCausalLM(pad_token_id=0),
+)
+trainer.train()
+```
+
+### KOLMOG — True Checkpoint Resume
+
+```python
+# v0.0.3: optimizer + scheduler state is saved, enabling true resume
+trainer.load_checkpoint("runs/my_run/checkpoint-5000")
+trainer.train()  # continues from exact state
+```
+
+### OMG — Diffusion LM
+
+```python
+from kolmgformers import OMGConfig, OMGModel
+
+config = OMGConfig(vocab_size=32000, hidden_size=768, num_layers=12)
+model  = OMGModel(config)
+
+import torch
+prompt = torch.tensor([[1, 42]])
+out = model.generate(prompt, new_tokens=128, steps=10)
+```
+
+---
+
+## Architecture: KOLMOG
+
+Based on the Kolmogorov-Arnold representation theorem:
+
+```
+F(X) = Σ_q Φ_q( Σ_i φ_{q,i}( xᵢ ⊕ eᵢ ⊕ c_{q,i} ) )
+```
+
+Key innovations:
+- **KAN layers** — learnable B-spline activations per edge (not fixed non-linearities)
+- **NCE** — Normalized Context Extraction: jackknife leave-one-out mean context
+- **WCC** — Weighted Cumulative Context: attention-like token selectivity at O(n·d)
+- **PLKAN** — Piecewise Linear KAN: 3–5× faster than B-spline, same expressivity
+- **No attention** — O(n·d) time, O(d) memory (independent of sequence length)
+
+---
+
+## Architecture: OMG
+
+Parallel Diffusion Language Model with:
+- GQA / MLA / Sliding-Window / Linear / Block-Sparse attention
+- MoE (dense + soft MoE)
+- DS-PDLM dual-stream (understanding + generation)
+- LoRA / DoRA PEFT
+- TASA + MFS + DI efficiency trilogy
+- TWE temporal embeddings, NCA neuro-creative routing
+
+---
+
+## Bug Fix History
+
+| Version | Fixes |
+|---------|-------|
+| v0.0.1 | #K1–#K4 (config, imports, KANLayer rightmost knot) |
+| v0.0.2 | WCC + PLKANLayer added |
+| v0.0.3 | #K5–#K15 (context mask, generate, PLKAN, phi layers, causal prefix, gradient checkpointing) + #T1–#T7 (early stopping, optimizer save/load, bf16 CPU, special token masking) |
+
+---
+
+## License
+
+Apache-2.0

kolmgformers-0.0.3/README.md

@@ -0,0 +1,184 @@
+# KOLMGformers  v0.0.3
+
+Unified Python library merging two research-grade model families:
+
+| Family | Architecture | Key property |
+|--------|-------------|-------------|
+| **KOLMOG** | Kolmogorov-Arnold Cumulative Context | Attention-free, O(d) memory |
+| **OMG** | Parallel Diffusion Transformer | Masked diffusion, full feature set |
+
+---
+
+## What's New in v0.0.3 — Bug Fixes & Improvements
+
+### Bug Fixes (KOLMOG)
+
+| ID | Component | Issue | Fix |
+|----|-----------|-------|-----|
+| #K5 | `KOLMOGformerLayer` | `phi` received raw `kappa_out` and `context` — shape contract was implicit and fragile | Documented and type-checked; context extractor now propagates `attention_mask` |
+| #K6 | `generate()` | Repetition penalty used a Python loop over `set(generated[b].tolist())` — O(seq·vocab) per step | Vectorised with `tensor.unique()` + `scatter_` |
+| #K7 | `generate()` | Top-p nucleus sampling: `cumprobs − softmax(sorted)` double-subtracted the pivot token, causing off-by-one exclusions | Replaced with correct shifted-cumsum implementation |
+| #K8 | `PLKANLayer` | `breakpoints` initialized via `expand().clone()` left non-contiguous memory; subtle autograd issues under in-place ops | Replaced with `linspace(...).repeat()` → always contiguous |
+| #K9 | `InnerKolmogorovFunction` | Always used slow B-spline `KANLayer` for φ layers, ignoring `config.use_plkan` | `build_kan_layer` factory now honoured for φ too (~3–5× speedup with PLKAN) |
+| #K10 | `CumulativeContextExtractor` | Causal pad+shift produced wrong exclusive prefix at position 0 (current token leaked into its own context) | Replaced with correct exclusive cumsum: `C^{<i} = cumsum[i] − kappa_w[i]` |
+| #K11 | `CumulativeContextExtractor` | `attention_mask` was accepted by model but never threaded to the context extractor; pad tokens polluted context vectors | Mask is now applied to `kappa` before accumulation at every layer |
+| #K12 | `KOLMOGformerForCausalLM` | Logits returned only `[:, :-1, :]` (already shifted), breaking downstream use of the full logit tensor | Full-sequence logits returned; shift applied only inside loss computation (matches HF API) |
+| #K13 | `save_pretrained` | Direct `torch.save` to final path could leave a corrupted checkpoint on interruption | Atomic write via `tempfile` + `os.replace`; safetensors support added |
+| #K14 | `KOLMOGformerModel` | No gradient checkpointing — OOM on long sequences during training | `enable_gradient_checkpointing()` added; controlled via `TrainingArguments.gradient_checkpointing` |
+| #K15 | `KANLayer.b_splines` | Grid buffer could be float32 while activations are bfloat16/float16, causing dtype mismatch | Grid is cast to `x.dtype` on every forward pass |
+
+### Bug Fixes (Training)
+
+| ID | Component | Issue | Fix |
+|----|-----------|-------|-----|
+| #T1 | `Trainer` | No early stopping — training continued even after convergence | `early_stopping_patience` added to `TrainingArguments` |
+| #T2 | `DataCollatorForCausalLM` | Sequences silently truncated or accepted without warning | `warn_length` parameter warns when batch sequences exceed model's position limit |
+| #T3 | `Trainer._save_checkpoint` | Saved only model weights — optimizer/scheduler lost, training couldn't truly resume | Optimizer + scheduler state saved in `trainer_state.pt` |
+| #T4 | `Trainer.load_checkpoint` | Restored only model weights and step count | Now restores optimizer, scheduler, early-stopping state |
+| #T5 | `get_scheduler` | Missing `"constant_with_warmup"` type | Added |
+| #T6 | `Trainer` | `bf16=True` on CPU silently fell back to float32 with no warning | Warning emitted; autocast errors caught gracefully |
+| #T7 | `DataCollatorForMaskedLM` | Random-replacement tokens were drawn from `[0, vocab_size)` including `[PAD]`/`[BOS]`/`[EOS]` | Now draws from `[num_special_tokens, vocab_size)` |
+
+### New Features (v0.0.3)
+
+**`KOLMOGformerConfig` additions:**
+- `context_dropout` — independent dropout on the context vector path (default `0.0`).
+- `ffn_type` — FFN activation: `"gelu"` (default) | `"silu"` | `"swiglu"`.
+- `max_position_embeddings_dynamic` — RoPE cache auto-extends beyond limit instead of erroring (default `True`).
+- `validate()` — called in `__post_init__`; surfaces config errors early with helpful messages.
+- `__repr__` — readable summary of key config fields.
+
+**`TrainingArguments` additions:**
+- `early_stopping_patience` — stop after N evaluations without improvement.
+- `gradient_checkpointing` — enable memory-efficient training automatically.
+
+---
+
+## Installation
+
+```bash
+pip install -e .
+# Optional extras
+pip install -e ".[hf]"       # HuggingFace tokenizers
+pip install -e ".[flash]"    # Flash Attention 2
+pip install -e ".[all]"      # Everything
+```
+
+---
+
+## Quick Start
+
+### KOLMOG — Attention-Free Causal LM
+
+```python
+from kolmgformers import KOLMOGformerConfig, KOLMOGformerForCausalLM
+import torch
+
+config = KOLMOGformerConfig(
+    vocab_size=32000,
+    hidden_size=512,
+    num_channels=8,
+    num_layers=6,
+    causal=True,
+    use_nce=True,    # Normalized Context Extraction
+    use_wcc=True,    # Weighted Cumulative Context (v0.0.2+)
+    use_plkan=True,  # Piecewise Linear KAN — 3-5x faster (v0.0.2+)
+)
+model = KOLMOGformerForCausalLM(config)
+print(config)  # v0.0.3: readable repr
+
+ids = torch.tensor([[1, 42, 100]])
+out = model.generate(ids, max_new_tokens=50, temperature=0.8)
+```
+
+### KOLMOG — Training with Early Stopping
+
+```python
+from kolmgformers import (
+    KOLMOGTrainer, KOLMOGTrainingArguments,
+    KOLMOGDataCollatorForCausalLM,
+)
+
+args = KOLMOGTrainingArguments(
+    output_dir="runs/my_run",
+    num_train_epochs=10,
+    early_stopping_patience=3,   # v0.0.3: stop after 3 bad evals
+    gradient_checkpointing=True, # v0.0.3: save memory on long seqs
+    evaluation_strategy="steps",
+    eval_steps=500,
+)
+trainer = KOLMOGTrainer(
+    model=model,
+    args=args,
+    train_dataset=train_ds,
+    eval_dataset=val_ds,
+    data_collator=KOLMOGDataCollatorForCausalLM(pad_token_id=0),
+)
+trainer.train()
+```
+
+### KOLMOG — True Checkpoint Resume
+
+```python
+# v0.0.3: optimizer + scheduler state is saved, enabling true resume
+trainer.load_checkpoint("runs/my_run/checkpoint-5000")
+trainer.train()  # continues from exact state
+```
+
+### OMG — Diffusion LM
+
+```python
+from kolmgformers import OMGConfig, OMGModel
+
+config = OMGConfig(vocab_size=32000, hidden_size=768, num_layers=12)
+model  = OMGModel(config)
+
+import torch
+prompt = torch.tensor([[1, 42]])
+out = model.generate(prompt, new_tokens=128, steps=10)
+```
+
+---
+
+## Architecture: KOLMOG
+
+Based on the Kolmogorov-Arnold representation theorem:
+
+```
+F(X) = Σ_q Φ_q( Σ_i φ_{q,i}( xᵢ ⊕ eᵢ ⊕ c_{q,i} ) )
+```
+
+Key innovations:
+- **KAN layers** — learnable B-spline activations per edge (not fixed non-linearities)
+- **NCE** — Normalized Context Extraction: jackknife leave-one-out mean context
+- **WCC** — Weighted Cumulative Context: attention-like token selectivity at O(n·d)
+- **PLKAN** — Piecewise Linear KAN: 3–5× faster than B-spline, same expressivity
+- **No attention** — O(n·d) time, O(d) memory (independent of sequence length)
+
+---
+
+## Architecture: OMG
+
+Parallel Diffusion Language Model with:
+- GQA / MLA / Sliding-Window / Linear / Block-Sparse attention
+- MoE (dense + soft MoE)
+- DS-PDLM dual-stream (understanding + generation)
+- LoRA / DoRA PEFT
+- TASA + MFS + DI efficiency trilogy
+- TWE temporal embeddings, NCA neuro-creative routing
+
+---
+
+## Bug Fix History
+
+| Version | Fixes |
+|---------|-------|
+| v0.0.1 | #K1–#K4 (config, imports, KANLayer rightmost knot) |
+| v0.0.2 | WCC + PLKANLayer added |
+| v0.0.3 | #K5–#K15 (context mask, generate, PLKAN, phi layers, causal prefix, gradient checkpointing) + #T1–#T7 (early stopping, optimizer save/load, bf16 CPU, special token masking) |
+
+---
+
+## License
+
+Apache-2.0