Skip to main content

ckg-nvidia-nemotron

Compressed Knowledge Graph · NVIDIA Nemotron 3 Ultra 550B A55B

71 nodes · 120+ edges · architecture, training pipeline, reasoning modes, deployment, benchmarks, agentic integrations.

Traversable via MCP — agents query declared relationships instead of inferring them.

pip install ckg-nvidia-nemotron

What's in the graph

Category Concepts
Architecture Hybrid Mamba-Transformer MoE, LatentMoE, MTP layers, 1M context
Training 20T tokens (NVFP4), 7 stages: Pretrain → SFT → RLVR → MOPD → MTP Boosting
RLVR DAPO (NeMo RL), GRPO (NeMo Gym), 6 reward domains
MOPD 10+ teacher models, 2 iterative distillation rounds
Reasoning Off / Regular / Medium + Reasoning Budget Control
Deployment vLLM, SGLang, TensorRT-LLM, NIM Microservices
Hardware DGX Cluster (4-node), GB300 DGX Station (selective expert offloading)
Parallelism TP=2, PP=6/12, EP=32
Fine-tuning LoRA via NeMo AutoModel, Megatron-Bridge, Unsloth
Agentic Hermes Agent, OpenClaw, tool use, multi-step planning, deep research
Benchmarks Intelligence Index 47.7, IFBench 81.4%, Ruler 95%, PinchBench 91%, Terminal-Bench 54%
Pricing $0.50/$2.20 per 1M tokens (NIM API)

MCP tools

  • search_concepts(query) — keyword search across all 70 nodes
  • query_ckg(concept, depth) — subgraph traversal (1–3 hops)
  • get_prerequisites(concept) — full upstream dependency chain
  • list_domains() — package metadata
  • get_benchmark_summary() — scores, competitive positioning, pricing
  • get_deployment_guide(target) — nim / vllm / sglang / tensorrt / fine-tuning

Usage

Claude Desktop / Cursor / Claude Code

{
  "mcpServers": {
    "ckg-nvidia-nemotron": {
      "command": "uvx",
      "args": ["ckg-nvidia-nemotron"]
    }
  }
}

Python

from ckg_nvidia_nemotron.server import search_concepts, query_ckg, get_prerequisites

# Find the LatentMoE routing architecture
print(search_concepts("LatentMoE"))

# Traverse training pipeline
print(query_ckg("MOPD", depth=2))

# What must be in place before reasoning modes work?
print(get_prerequisites("Reasoning Mode Medium"))

HTTP server

ckg-nvidia-nemotron --port 8000
# MCP endpoint: POST http://localhost:8000/mcp

Benchmark

Grounded in the Compact Knowledge Graph benchmark (v0.6.2).

EVAL

benchmark: ckg-benchmark v0.6.2
dataset: huggingface.co/datasets/danyarm/ckg-benchmark
benchmarked: false
rag_baseline_f1: 0.123
graphrag_baseline_f1: 0.120
mean_tokens: 269
paper: github.com/Yarmoluk/ckg-benchmark/blob/main/paper/main.pdf

META

version: 0.1.1
domain: nvidia-nemotron
nodes: 71
edges: 120+
model_covered: Nemotron 3 Ultra 550B A55B
nvidia_release_date: 2026-06-04
license_of_model: OpenMDW-1.1
package_license: MIT
publisher: Graphify.md
publisher_url: https://graphifymd.com/pro/

Built by Graphify.md · Benchmark paper · patent pending

Metadata

Release files for ckg-nvidia-nemotron 0.2.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for ckg-nvidia-nemotron 0.2.0
File Size Uploaded Provenance
ckg_nvidia_nemotron-0.2.0.tar.gz 13.4 kB Without provenance Details

Built distribution (wheel)

Table of built distributions (wheels) for ckg-nvidia-nemotron 0.2.0
File Interpreter ABI Platform
ckg_nvidia_nemotron-0.2.0-py3-none-any.whl Python 3 none any Details

Total release size: 29.9 kB

Release files / ckg_nvidia_nemotron-0.2.0.tar.gz

Download URL ckg_nvidia_nemotron-0.2.0.tar.gz
Size 13.4 kB
Tags Source
SHA-256 checksum
How to use checksums
dd75ac2948cbbff23f7baae012fba6aed12b2846e3fe826d5afe934018728161
BLAKE2b-256 checksum
How to use checksums
6368cf393c76c975c318455e9bbff37598385523a38b5a32cc1b1a71e6dc971a
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.11.14

Release files / ckg_nvidia_nemotron-0.2.0-py3-none-any.whl

Download URL ckg_nvidia_nemotron-0.2.0-py3-none-any.whl
Size 16.5 kB
Tags Python 3
SHA-256 checksum
How to use checksums
34dbb7a8b7839e76210fe1c70bd8f8bbe033c24ed66994e6a88011e736c44b1a
BLAKE2b-256 checksum
How to use checksums
4cb655f888d5a665b996dc5e9573f794cc1b871981f5f092b8254dccfd0abd5d
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.11.14

Security for ckg-nvidia-nemotron 0.2.0

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

0.2.0 This release

2 release files No provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page