Views
No views yet

1uv pip install git+https://github.com/vllm-project/vllm.git
2uv pip install llmcompressor1vllm serve RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block \
2 --host 0.0.0.0 --port 8088 \
3 --tensor-parallel-size 8 \
4 --enable-expert-parallel \
5 --max-model-len 262144 \
6 --gpu-memory-utilization 0.90 \
7 --max-num-seqs 32 \
8 --max-num-batched-tokens 32768 \
9 --enable-chunked-prefill \
10 --enable-prefix-caching \
11 --reasoning-parser nemotron_v3 \
12 --mamba-ssm-cache-dtype float16 \
13 --mamba-backend flashinfer \
14 --enable-mamba-cache-stochastic-rounding \
15 --mamba-cache-philox-rounds 5 \
16 --speculative-config '{"method": "nemotron_h_mtp", "num_speculative_tokens": 5}' \
17 --model-loader-extra-config '{"enable_multithread_load": true, "num_threads": 96}' \
18 --trust-remote-code1from openai import OpenAI
2
3openai_api_key = "EMPTY"
4openai_api_base = "http://localhost:8088/v1"
5
6client = OpenAI(
7 api_key=openai_api_key,
8 base_url=openai_api_base,
9)
10
11model = "RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block"
12
13messages = [
14 {"role": "user", "content": "Solve for x: 2x + 5 = 13"},
15]
16
17outputs = client.chat.completions.create(
18 model=model,
19 messages=messages,
20)
21
22generated_text = outputs.choices[0].message.content
23print(generated_text)1from llmcompressor import model_free_ptq
2
3MODEL_ID = "nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16"
4SAVE_DIR = MODEL_ID.rstrip("/").split("/")[-1] + "-FP8-block"
5
6model_free_ptq(
7 model_stub=MODEL_ID,
8 save_directory=SAVE_DIR,
9 scheme="FP8_BLOCK",
10 ignore=[
11 "re:.*gate$",
12 "lm_head",
13 "model.embed_tokens",
14 "re:.*mixer.conv1d.*",
15 "re:.*norm_f*",
16 "re:.*bias$",
17 "re:.*embed_tokens$",
18 "backbone.embeddings"
19 ],
20 max_workers=15,
21 device="cuda:0",
22)1uv pip install git+https://github.com/vllm-project/vllm.git
2uv pip install lighteval==0.13.0
3uv pip install "litellm[caching]>=1.66.0"1vllm serve RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block \
2 --host 0.0.0.0 --port 8088 \
3 --tensor-parallel-size 8 \
4 --enable-expert-parallel \
5 --max-model-len 262144 \
6 --gpu-memory-utilization 0.90 \
7 --max-num-seqs 32 \
8 --max-num-batched-tokens 32768 \
9 --enable-chunked-prefill \
10 --enable-prefix-caching \
11 --reasoning-parser nemotron_v3 \
12 --mamba-ssm-cache-dtype float16 \
13 --mamba-backend flashinfer \
14 --enable-mamba-cache-stochastic-rounding \
15 --mamba-cache-philox-rounds 5 \
16 --speculative-config '{"method": "nemotron_h_mtp", "num_speculative_tokens": 5}' \
17 --model-loader-extra-config '{"enable_multithread_load": true, "num_threads": 96}' \
18 --trust-remote-code1lighteval endpoint litellm \
2 "model_name=hosted_vllm/RedHatAI__NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block,provider=hosted_vllm,base_url=http://127.0.0.1:8088/v1,timeout=3600,concurrent_requests=32,generation_parameters={temperature:1.0,top_p:0.95,max_new_tokens:32768}" \
3 "aime25|0" \
4 --output-dir results --save-details1lighteval endpoint litellm \
2 "model_name=hosted_vllm/RedHatAI__NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block,provider=hosted_vllm,base_url=http://127.0.0.1:8088/v1,timeout=3600,concurrent_requests=32,generation_parameters={temperature:1.0,top_p:0.95,max_new_tokens:32768}" \
3 "gpqa:diamond|0" \
4 --output-dir results --save-details| Benchmark | nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16 | nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-NVFP4 | RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-dynamic | RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block (this model) | RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-quantized.w4a16 |
|---|---|---|---|---|---|
| AIME 2025 (pass@1) | 90.00 | 90.00 (100.0%) | 93.33 (103.7%) | 86.67 (96.3%) | 86.67 (96.3%) |
| GPQA Diamond (pass@1) | 78.79 | 84.85 (107.7%) | 82.32 (104.5%) | 81.31 (103.2%) | 81.82 (103.8%) |
| Average | 84.39 | 87.42 (103.6%) | 87.83 (104.1%) | 83.99 (99.5%) | 84.24 (99.8%) |
| Category | Benchmark | nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16 | RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block (this model) | Recovery |
|---|---|---|---|---|
| Instruct | MMLU-CoT (5-shot) | 91.09 | 90.98 | 99.88% |
| Instruct | GSM8K Platinum (5-shot) | 98.59 | 98.68 | 100.09% |
| Instruct | IFEval (0-shot) | 91.00 | 90.70 | 99.67% |
| Instruct | MATH-500 | 83.60 | 82.80 | 99.04% |
| Reasoning | AIME 2025 | 64.17 | 65.42 | 101.95% |
| Reasoning | MATH-500 | 85.80 | 86.00 | 100.23% |
| Reasoning | GSM8K Platinum (0-shot) | 96.77 | 97.08 | 100.32% |
| Reasoning | IFEval (0-shot) | 94.58 | 94.82 | 100.25% |
| Coding | LCB CodeGen v6 | 51.81 | 52.57 | 101.47% |
| Category | Benchmark | nvidia/NVIDIA-Nemotron-3-Ultra-550B-A55B-BF16 | RedHatAI/NVIDIA-Nemotron-3-Ultra-550B-A55B-FP8-block (this model) | Recovery |
|---|---|---|---|---|
| Overall | BFCLv4 Overall Acc | 55.44 | 54.84 | 98.92% |
| Single Turn | Non-Live Acc | 45.00 | 44.31 | 98.47% |
| Single Turn | Live Acc | 71.65 | 72.32 | 100.94% |
| Multi-Turn | Multi-Turn Acc | 42.12 | 42.75 | 101.50% |
| Agentic | Agentic Acc | 57.64 | 55.53 | 96.34% |