feat(trtllm): support agentic training for SWE rollout - #3130

Merged

terrykong merged 27 commits into

NVIDIA-NeMo:mainfrom

hchings:trtllm-agentic-swe

Jul 26, 2026

Commits on Jul 25, 2026

add trtllm agentic scripts
hchings
committed
9e64894
4b smoke e2e for trtllm (agentic): prompt_token_ids + max_input_len fixes, agentic Dockerfile, 4b smoke script
hchings
committed
31bd5f3
30B 16 nodes trtllm swe
hchings
committed
7332f05
fix multiturn, fix env
hchings
committed
5e4a75c
fix empty gen token case
hchings
committed
f670c1e
refactor
hchings
committed
9b66bdf
rebase fix
hchings
committed
eb70fc7
cleanup
hchings
committed
3e4b271
refactor into utils, cleanup
hchings
committed
f1e37de
reconcile trtllm http-server piping post-rebase
hchings
committed
7cff099
address comments
hchings
committed
00120fc
address comments, pre-commit
hchings
committed
267a288
multinode rollout for non colocated, allow quantized ckpt
hchings
committed
6a94e04
cross-node rollout + tool parser refactor and fixes
hchings
committed
a249532
qwen3.5b refit
hchings
committed
6a2bf93
address comments (reasoning parsers, eos handling, tests)
hchings
committed
0591b0e
trtllm: nsys profiling of the generation (rollout) GPU workers

Superjomn
authored and
hchings
committed
cafbc4b
docs: nsys profiling for TRT-LLM generation (rollout) workers

Superjomn
authored and
hchings
committed
a405f51
fix reasoning parsing
hchings
committed
64dd27f
fix: satisfy lint checks

joyang-nv
authored and
hchings
committed
0a96d6c
fix: include OpenAI server utils in Pyrefly

joyang-nv
authored and
hchings
committed
11b21dc
nit & address comments
hchings
committed
bfef605
Pyrefly fix
hchings
committed
7c33404
fix vllm failed test to unblock CI
hchings
committed
e62fb40
Merge branch 'main' into trtllm-agentic-swe
joyang-nv
authored
bffcfbe
make import of new trtllm refit api optional
hchings
committed
6ee8f61
remove trtllm script for 30B-Thinking; add back in another MR
hchings
committed
07dbfd4