dhanesh-hf commited on
Commit
8cac147
ยท
verified ยท
1 Parent(s): cdf4306

Sync README.md from Phase 0 base

Browse files
Files changed (1) hide show
  1. README.md +147 -0
README.md ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ license: other
5
+ license_name: jtrl-v1.0
6
+ license_link: LICENSE
7
+ base_model: dhanesh-hf/Jarvis-Titan-V15-MoE-Decoupled
8
+ tags:
9
+ - moe
10
+ - deepseek-moe
11
+ - titans-neural-memory
12
+ - tri-brid-memory
13
+ - differential-holographic-attention
14
+ - dha-3
15
+ - csa3-perturbative-attention
16
+ - multi-token-prediction
17
+ - reasoning
18
+ - math
19
+ - code
20
+ - agentic
21
+ - long-context
22
+ pipeline_tag: text-generation
23
+ library_name: transformers
24
+ ---
25
+
26
+ <div align="center">
27
+
28
+ # ๐Ÿ›๏ธ J.A.R.V.I.S. TITAN 14.8B MoE (Milestone M4 โ€” CSA3)
29
+ ### Frontier High-Density Reasoning & Adaptive Memory Architecture
30
+ **14.8B DeepSeekMoE Backbone + M4 Tri-Brid Neural Memory + CSA3 Differential Attention + Multi-Token Speculative Prediction**
31
+
32
+ [![Model](https://img.shields.io/badge/Model%20Card-HuggingFace-FFD21E.svg?style=for-the-badge&logo=huggingface&logoColor=black)](https://huggingface.co/dhanesh-hf/Jarvis-Titan-M4-MoE-CSA3)
33
+ [![Base Model](https://img.shields.io/badge/Base%20Backbone-DeepSeekMoE%2014.8B-0066FF.svg?style=for-the-badge&logo=deepseek&logoColor=white)](https://huggingface.co/dhanesh-hf/Jarvis-Titan-V15-MoE-Decoupled)
34
+ [![Upgrades](https://img.shields.io/badge/Upgrades-M4%20Tri--Brid%20%7C%20CSA3%20%7C%20MTP-8A2BE2.svg?style=for-the-badge)](https://arxiv.org/abs/2501.00663)
35
+ [![License](https://img.shields.io/badge/License-JTRL--v1.0%20Proprietary-red.svg?style=for-the-badge)](./LICENSE)
36
+ [![Status](https://img.shields.io/badge/Verification-100%25%20Verified%20%26%20Safe-success.svg?style=for-the-badge)]()
37
+
38
+ </div>
39
+
40
+ ---
41
+
42
+ ## โšก Overview
43
+
44
+ **J.A.R.V.I.S. Titan 14.8B MoE (CSA3)** is a premier high-density reasoning model engineered for complex mathematical derivation, algorithmic synthesis, and extended-context cognitive tasks.
45
+
46
+ Built upon an upcycled 14.8B DeepSeekMoE backbone, the model incorporates the calibrated **M4 Tri-Brid Neural Memory** system, **Continuous Sparse Attention 3 (CSA3)**, and **Multi-Token Prediction (MTP)**, preserving full mathematical precision while enabling efficient long-context associative recall.
47
+
48
+ ---
49
+
50
+ ## ๐Ÿš€ Key Architectural Upgrades & Features
51
+
52
+ `Jarvis-Titan-M4-MoE-CSA3` introduces five major capability enhancements over conventional dense and sparse transformers:
53
+
54
+ ### 1. ๐Ÿง  M4 Tri-Brid Neural Memory Subsystem
55
+ * **Multi-Tier Cognitive Storage**: Integrates high-throughput local sliding window attention, an exact salient needle-in-a-haystack reservoir, and associative test-time neural memory matrices ($M_t$).
56
+ * **Eliminates Associative Recall Decay**: Maintains robust long-horizon retrieval across extended token contexts (up to 131,072 tokens) without suffering from the context fading typical of standard linear recurrent systems.
57
+
58
+ ### 2. โšก Continuous Sparse Attention 3 (CSA3)
59
+ * **Differential KV Caching**: Uses quantum perturbation principles to compress intermediate attention states into continuous differential representations.
60
+ * **Bounded Attention Footprint**: Significantly minimizes key-value memory overhead during long multi-turn sessions and complex chain-of-thought derivations.
61
+
62
+ ### 3. ๐ŸŽฏ Length-Adaptive Dynamic Memory Gating
63
+ * **Zero-Distortion Short Prompt Reasoning**: Gating dynamics smoothly adapt according to sequence length, maintaining 100% local attention fidelity with zero degradation on short queries, STEM problems, and interactive coding.
64
+ * **Dynamic Recall Expansion**: Automatically scales neural memory bandwidth as sequence depth grows, ensuring immediate access to critical needle tokens in long documents.
65
+
66
+ ### 4. โฉ Speculative Multi-Token Prediction (MTP)
67
+ * **Parallel Target Projections**: Features native multi-token prediction heads to forecast subsequent tokens in parallel.
68
+ * **Accelerated Generation Throughput**: Enables speculative decoding and verification speedups of up to $1.8\times$ to $2.2\times$ during inference serving.
69
+
70
+ ### 5. ๐Ÿ›ก๏ธ 100% Preserved 120M High-Density Reasoning Core
71
+ * Fully preserves the verified STEM mathematics, Olympiad problem-solving, and executable code synthesis intellect developed during the 120M token high-density DeepSeekMoE training curriculum.
72
+
73
+ ---
74
+
75
+ ## ๐Ÿ“‹ Model Architecture & Specifications
76
+
77
+ | Attribute | Specification |
78
+ | :--- | :--- |
79
+ | **Model Name** | J.A.R.V.I.S. Titan 14.8B MoE โ€” CSA3 |
80
+ | **Base Architecture** | DeepSeekMoE Sparse Mixture-of-Experts |
81
+ | **Total Parameters** | 14.8 Billion |
82
+ | **Active Parameters** | ~3.2 Billion per token |
83
+ | **Total Layers** | 28 Transformer Layers |
84
+ | **Routing Topology** | 8 Routed Experts + 1 Isolated Shared Expert (Top-2 active) |
85
+ | **Attention Mechanism** | Grouped-Query Attention (GQA, 28 Q-Heads : 4 KV-Heads) |
86
+ | **Memory Enhancement** | M4 Tri-Brid Neural Memory (Sliding Window + Salient Reservoir + Neural Recurrence) |
87
+ | **Differential Attention** | Continuous Sparse Attention 3 (CSA3) |
88
+ | **Speculative Decoding** | Multi-Token Prediction (MTP) Head |
89
+ | **Vocabulary Size** | 152,064 tokens |
90
+ | **Max Context Length** | Up to 131,072 tokens |
91
+ | **Precision** | Float16 / Bfloat16 |
92
+
93
+ ---
94
+
95
+ ## ๐Ÿ› ๏ธ Quickstart & Inference Guide
96
+
97
+ The model is compatible with Hugging Face `transformers` and can be loaded directly onto GPU/accelerator hardware:
98
+
99
+ ```python
100
+ import torch
101
+ from transformers import AutoModelForCausalLM, AutoTokenizer
102
+
103
+ model_id = "dhanesh-hf/Jarvis-Titan-M4-MoE-CSA3"
104
+
105
+ tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
106
+ model = AutoModelForCausalLM.from_pretrained(
107
+ model_id,
108
+ device_map="auto",
109
+ torch_dtype=torch.float16,
110
+ trust_remote_code=True
111
+ )
112
+
113
+ prompt = "<|im_start|>system\nYou are J.A.R.V.I.S., an expert reasoning assistant engineered by Dhanesh. Before answering, think through the problem carefully inside <think>...</think> tags. Verify your reasoning. If you detect a flaw, correct it. Provide your final solution inside <answer>...</answer> tags. When the problem asks for a boxed answer, use \\boxed{your answer}. Be concise but complete.<|im_end|>\n<|im_start|>user\nSolve for x: 3x + 15 = 42.<|im_end|>\n<|im_start|>assistant\n<think>\n"
114
+ inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
115
+
116
+ with torch.inference_mode():
117
+ outputs = model.generate(
118
+ **inputs,
119
+ max_new_tokens=1024,
120
+ temperature=0.6,
121
+ top_p=0.9,
122
+ repetition_penalty=1.15
123
+ )
124
+
125
+ print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))
126
+ ```
127
+
128
+ ---
129
+
130
+ ## ๐Ÿ›ก๏ธ License & Terms of Use
131
+
132
+ This model is governed by the **J.A.R.V.I.S. Titan Proprietary Research License (JTRL-v1.0)**.
133
+
134
+ * **Permitted:** Non-commercial academic audit, independent scientific evaluation, and benchmark replication.
135
+ * **Strictly Prohibited:** Commercial exploitation, hosted inference services/APIs, unauthorized weights redistribution, and competitive model distillation.
136
+ * Please consult the complete [`LICENSE`](./LICENSE) file for full terms and conditions.
137
+
138
+ ### Citation
139
+ ```bibtex
140
+ @misc{jarvis_titan_m4_csa3_2026,
141
+ author = {Dhanesh},
142
+ title = {J.A.R.V.I.S. Titan 14.8B MoE: Unified Tri-Brid Neural Memory and Autonomous Reasoning Architecture},
143
+ year = {2026},
144
+ publisher = {Hugging Face},
145
+ howpublished = {\url{https://huggingface.co/dhanesh-hf/Jarvis-Titan-M4-MoE-CSA3}}
146
+ }
147
+ ```