CoolFace
Modelpublic

dhanesh-hf/Jarvis-Titan-M4-MoE-CSA3

sourceHugging Faceotherupdated 8d agoView on Hugging Face
0likes949downloads
Model Card

<div align="center">

๐Ÿ›๏ธ J.A.R.V.I.S. TITAN 14.8B MoE (Milestone M4 โ€” CSA3)

Frontier High-Density Reasoning & Adaptive Memory Architecture

14.8B DeepSeekMoE Backbone + M4 Tri-Brid Neural Memory + CSA3 Differential Attention + Multi-Token Speculative Prediction

![Model](https://huggingface.co/dhanesh-hf/Jarvis-Titan-M4-MoE-CSA3) ![Base Model](https://huggingface.co/dhanesh-hf/Jarvis-Titan-V15-MoE-Decoupled) ![Upgrades](https://arxiv.org/abs/2501.00663) ![License](./LICENSE) ![Status]()

</div>


โšก Overview

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.

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.


๐Ÿš€ Key Architectural Upgrades & Features

Jarvis-Titan-M4-MoE-CSA3 introduces five major capability enhancements over conventional dense and sparse transformers:

1. ๐Ÿง  M4 Tri-Brid Neural Memory Subsystem

  • โ€”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$).
  • โ€”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.

2. โšก Continuous Sparse Attention 3 (CSA3)

  • โ€”Differential KV Caching: Uses quantum perturbation principles to compress intermediate attention states into continuous differential representations.
  • โ€”Bounded Attention Footprint: Significantly minimizes key-value memory overhead during long multi-turn sessions and complex chain-of-thought derivations.

3. ๐ŸŽฏ Length-Adaptive Dynamic Memory Gating

  • โ€”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.
  • โ€”Dynamic Recall Expansion: Automatically scales neural memory bandwidth as sequence depth grows, ensuring immediate access to critical needle tokens in long documents.

4. โฉ Speculative Multi-Token Prediction (MTP)

  • โ€”Parallel Target Projections: Features native multi-token prediction heads to forecast subsequent tokens in parallel.
  • โ€”Accelerated Generation Throughput: Enables speculative decoding and verification speedups of up to $1.8\times$ to $2.2\times$ during inference serving.

5. ๐Ÿ›ก๏ธ 100% Preserved 120M High-Density Reasoning Core

  • โ€”Fully preserves the verified STEM mathematics, Olympiad problem-solving, and executable code synthesis intellect developed during the 120M token high-density DeepSeekMoE training curriculum.

๐Ÿ“‹ Model Architecture & Specifications

AttributeSpecification
Model NameJ.A.R.V.I.S. Titan 14.8B MoE โ€” CSA3
Base ArchitectureDeepSeekMoE Sparse Mixture-of-Experts
Total Parameters14.8 Billion
Active Parameters~3.2 Billion per token
Total Layers28 Transformer Layers
Routing Topology8 Routed Experts + 1 Isolated Shared Expert (Top-2 active)
Attention MechanismGrouped-Query Attention (GQA, 28 Q-Heads : 4 KV-Heads)
Memory EnhancementM4 Tri-Brid Neural Memory (Sliding Window + Salient Reservoir + Neural Recurrence)
Differential AttentionContinuous Sparse Attention 3 (CSA3)
Speculative DecodingMulti-Token Prediction (MTP) Head
Vocabulary Size152,064 tokens
Max Context LengthUp to 131,072 tokens
PrecisionFloat16 / Bfloat16

๐Ÿ› ๏ธ Quickstart & Inference Guide

The model is compatible with Hugging Face transformers and can be loaded directly onto GPU/accelerator hardware:

python
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "dhanesh-hf/Jarvis-Titan-M4-MoE-CSA3"

tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
    model_id,
    device_map="auto",
    torch_dtype=torch.float16,
    trust_remote_code=True
)

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"
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

with torch.inference_mode():
    outputs = model.generate(
        **inputs,
        max_new_tokens=1024,
        temperature=0.6,
        top_p=0.9,
        repetition_penalty=1.15
    )

print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))

๐Ÿ›ก๏ธ License & Terms of Use

This model is governed by the J.A.R.V.I.S. Titan Proprietary Research License (JTRL-v1.0).

  • โ€”Permitted: Non-commercial academic audit, independent scientific evaluation, and benchmark replication.
  • โ€”Strictly Prohibited: Commercial exploitation, hosted inference services/APIs, unauthorized weights redistribution, and competitive model distillation.
  • โ€”Please consult the complete `LICENSE` file for full terms and conditions.

Citation

bibtex
@misc{jarvis_titan_m4_csa3_2026,
  author = {Dhanesh},
  title = {J.A.R.V.I.S. Titan 14.8B MoE: Unified Tri-Brid Neural Memory and Autonomous Reasoning Architecture},
  year = {2026},
  publisher = {Hugging Face},
  howpublished = {\url{https://huggingface.co/dhanesh-hf/Jarvis-Titan-M4-MoE-CSA3}}
}