Dan-Carter / Affine-t3

Try K2-Think · K2 Technical Report

K2-Think is a 32 billion parameter open-weights general reasoning model with strong performance in competitive mathematical problem solving.

Quickstart

Transformers

You can use K2-Think with Transformers. If you use transformers.pipeline , it will apply the chat template automatically. If you use model.generate directly, you need to apply the chat template mannually.

from transformers import pipeline
import torch

model_id =  "LLM360/K2-Think"

pipe = pipeline(
    "text-generation",
    model=model_id,
    torch_dtype="auto",
    device_map="auto",
)

messages = [
    {"role": "user", "content": "what is the next prime number after 2600?"},
]

outputs = pipe(
    messages,
    max_new_tokens=32768,
)
print(outputs[0]["generated_text"][-1])

Evaluation & Performance

Detailed evaluation results are reported in out Tech Report

Benchmarks (pass@1, average over 16 runs)

Inference Speed

We deploy K2-THINK on Cerebras Wafer-Scale Engine (WSE) systems, leveraging the world’s largest processor and speculative decoding to achieve unprecedented inference speeds for our 32B reasoning system.

Safety Evaluation

Aggregated across four safety dimensions ( Safety-4 ):

Citation

@techreport{k2think2025,
  title   = {K2-Think: A Parameter-Efficient Reasoning System},
  author  = {Zhoujun Cheng and Richard Fan and Shibo Hao and Taylor W. Killian and Haonan Li and Suqi Sun and Hector Ren and Alexander Moreno and Daqian Zhang and Tianjun Zhong and Yuxin Xiong and Yuanzhe Hu and Yutao Xie and Xudong Han and Yuqi Wang and Varad Pimpalkhute and Yonghao Zhuang and Aaryamonvikram Singh and Xuezhi Liang and Anze Xie and Jianshu She and Desai Fan and Chengqian Gao and Liqun Ma and Mikhail Yurochkin and John Maggs and Xuezhe Ma and Guowei He and Zhiting Hu and Zhengzhong Liu and Eric P. Xing},
  year    = {2025},
  institution = {Institute of Foundation Models, Mohamed bin Zayed University of Artificial Intelligence},
  url     = {https://k2think-about.pages.dev/assets/tech-report/K2-Think_Tech-Report.pdf}
}