tmax-sft-8b

Image: image

💻 Code · 🤗 Models & Data · 📜 Paper · 📓 Blog

Note. For full information, go check out the Tmax paper here.

TMax 8B#

TMax 8B is a model trained using SFT on top of Qwen 3 8B for use as a terminal-agent. It was used as a base for the RL training of Tmax 8B.

This model is part of a collection of terminal agents in various sizes.

Evaluation Results#

For details on evaluation methodology please check our paper. In general, we used a podman (docker) backend with default timeouts and custom harness similar to mini-swe-agent.

Model Details#

Model Description#

• Developed by: Ai2
• Language(s) (NLP): English
• License: Apache 2.0
• Finetuned from model: Qwen 3 8B
• Dataset: TMax-15k

Use#

To use this model, we recommend serving with vllm (or your inference framework of choice) with:

uvx vllm==0.19.1 serve allenai/tmax-8b \
  --served-model-name tmax-8b \
  --enable-auto-tool-choice \
  --tool-call-parser qwen3_xml \
  --port 8008 \
  --max-model-len 40960 \
  --tensor-parallel-size 8 \
  --language_model_only

Make sure to set language_model_only as we removed the vision head during training.

For more details on evaluation, please see our codebase.

Hyperparameters#

This model was trained using SFT with the following hyperparameters:

• base model: Qwen/Qwen3-8B
• Dataset: tmax SFT
• Max overall tokens: 32768
• Global train batch size: 128
• Epochs: 2
• Learning rate: 2e-5
• LR scheduler: 0.03% warmup, linear cooldown

For more details on training, please see our codebase.

License#

This model is licensed under Apache 2.0. It is intended for research and educational use in accordance with Ai2's Responsible Use Guidelines.

Citation#

If you use our model or data, please cite our paper:

@misc{ivison2026tmaxsimplerecipeterminal,
      title={Tmax: A simple recipe for terminal agents}, 
      author={Hamish Ivison and Junjie Oscar Yin and Rulin Shao and Teng Xiao and Nathan Lambert and Hannaneh Hajishirzi},
      year={2026},
      eprint={2606.23321},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2606.23321}, 
}