DM-JEPA 1.1: Decision-Making Joint Embedding Predictive Architecture

Developed by Danger Labs
Ultra-fast, Non-Autoregressive System 1 Decision Model with 16k Extended Context

License: MIT Decision Index


⚑ Overview

DM-JEPA 1.1 is a non-autoregressive System 1 Decision Engine built upon Yann LeCun's Joint Embedding Predictive Architecture (JEPA) principles. Rather than generating textual Chain-of-Thought (CoT) tokens autoregressively, DM-JEPA conducts deliberate multi-step reasoning entirely within latent thought space.

  • Sub-35ms Latency: Delivers 34.3 ms median inference latency per decision request, compared to 1,000ms–8,000ms for standard 7B–70B autoregressive LLMs (~28Γ— faster).
  • Extended 16k Context: Supports up to 16,384 tokens of state context and 512 tokens per decision criterion, answering 99.84% (150,518 / 150,759) of the Decision Index full panel.
  • Zero Token Hallucination: Directly predicts compatibility energy between environmental state facts and candidate decision criteria without free-form token hallucination.
  • Listwise Joint Decision Space: Simultaneously evaluates arbitrary sets of decision options in a single forward pass with inter-candidate self-attention and cross-attentive fact verification.
  • Official Decision Index Full-Panel Score: 23.16 Balanced Skill / 41.92 Balanced Raw across all 150,759 requests of the full Decision Index suite (view evaluation dataset).

πŸ”¬ Architecture

flowchart TD
    State["Environment State & Instructions (X)"] --> StateEnc["ModernBERT Context Encoder (RoPE Scaled to 16k)"]
    Options["Candidate Criteria Options (Y_1 ... Y_K)"] --> CritEnc["Target Projection Criteria Encoder"]
    
    StateEnc --> HState["Contextual Sequence Embeddings h_state"]
    CritEnc --> SOpt["Candidate Target Embeddings s_options"]
    
    HState & SOpt --> Predictor["M-Step Latent Predictive Verifier (GRU-Gated Thought Rollout)"]
    Predictor --> SHat["Anticipated Latent Target Vector s_hat"]
    
    SHat & SOpt --> Scorer["Latent Compatibility Scorer (Cosine Similarity * Temperature)"]
    Scorer --> Probs["Decision Probabilities P(Y_k | X)"]
  1. State Context Encoder: Built upon answerdotai/ModernBERT-base with dynamic RoPE base frequency expansion supporting 16,384 tokens.
  2. Criteria Target Encoder: Computes semantic anchor embeddings for each candidate choice.
  3. M-Step Recurrent Latent Predictive Verifier: Performs recurrent steps of cross-attentive deduction in thought space conditioned on state facts.
  4. Calibrated Latent Scorer: Normalized inner product scoring with learnable calibration temperature to output calibrated probability distributions over options.

πŸ“Š Decision Index Full-Panel Performance (150,759 Requests)

Evaluated across the complete 150,759-request suite of the Decision Index (0.2.1 protocol):

Metric / Area Score Notes
Decision Index (Balanced Skill) 23.16 Full 44-benchmark panel (+26.1% over v1.0)
Raw Index (Balanced Raw) 41.92 Full panel
Breadth Skill 22.40 Geometric mean across all 5 domains
Tools & Automation 31.90% Skill 42.11% Raw (95.63% coverage)
Retrieval & Classification 28.23% Skill 42.58% Raw (100.00% coverage)
Knowledge & Reasoning 25.26% Skill 43.71% Raw (99.92% coverage)
Arts & Human Taste 16.73% Skill 38.67% Raw (100.00% coverage)
Language Understanding 13.44% Skill 40.81% Raw (100.00% coverage)

Head-to-Head Benchmark Highlights

Benchmark Domain Metric DM-JEPA 1.1 Result Leaderboard Champion (jev)
GSM8K (Arithmetic Reasoning) Raw Accuracy / Skill 100.00% / 100.00% 79.87% / 75.65%
Home Appliance Simulator Field Accuracy / Field Skill 91.12% / 82.25% 52.27% / 52.27%
OpenJev High-Trust Suite Raw Accuracy / Brier 100.00% / 0.0000 ~92.00%
Jevbench-Hard Raw Accuracy / Trap Avoidance 85.59% / 92.45% ~78.00%
Median Inference Latency Time per Request 34.3 ms 950 ms (28Γ— faster)

πŸš€ Quickstart

Installation

pip install torch transformers huggingface_hub safetensors

Loading the Model

from modeling_dm_jepa import DMJEPA
import torch

device = "cuda" if torch.cuda.is_available() else "cpu"
model = DMJEPA.from_pretrained("DangerLabs/DM-JEPA", device=device)

🏷️ Citation & Contact

@misc{dangerlabs2026dmjepa,
  author = {Danger Labs},
  title = {DM-JEPA 1.1: Non-Autoregressive System 1 Decision Model with Stretched Context and Latent Predictive Verification},
  year = {2026},
  publisher = {Hugging Face},
  howpublished = {\url{https://huggingface.co/DangerLabs/DM-JEPA}}
}

Published by Danger Labs Β· Hugging Face Organization

Downloads last month
-
Safetensors
Model size
0.3B params
Tensor type
F32
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support