simplex-ai-inc/LiteResearcher-4B-SFT

Model

LiteResearcher-4B-SFT

0

4 commits

3 linked in READMEs

updated Jun 9, 2026

See the code

README

LiteResearcher-4B-SFT

This is the SFT cold-start checkpoint for LiteResearcher β€” a scalable agentic RL training framework for deep-research agents.

It is the initial policy used to launch the two-stage curriculum RL training that produces the final simplex-ai-inc/LiteResearcher-4B model.

If you are looking for the final RL model, please use simplex-ai-inc/LiteResearcher-4B. If you want to reproduce the RL training from scratch, this is the checkpoint you need.

Model details

  • Base model: Qwen/Qwen3-4B-Thinking-2507
  • Architecture: Qwen3ForCausalLM (36 layers, hidden 2560, 32 heads, GQA 8 KV heads)
  • Max position embeddings: 262,144 (RoPE ΞΈ = 5,000,000)
  • Precision: bfloat16
  • Total params: ~4B
  • Training framework: LLaMA-Factory

Training recipe

ItemValue
StageSFT (cold-start before RL)
Base modelQwen/Qwen3-4B-Thinking-2507
Datasetsimplex-ai-inc/LiteResearcher-Data (~68.2k SFT trajectories)
Max sequence length64K (cutoff_len=65536)
Global batch size128 (per-device bs 2 Γ— grad-accum 8 Γ— 8 GPUs)
Epochs1
Optimizer steps533
Learning rate2.0e-5, cosine, 10% warmup
Final train lossβ‰ˆ 0.447 (starting loss β‰ˆ 1.19)

The SFT trajectories teach the model the ReAct think β†’ search β†’ visit β†’ answer loop and the strict <answer>...</answer> output contract used by the RL environment. Because the base is the Thinking-2507 variant, the model preserves long chain-of-thought behavior inside <think>...</think> blocks, which is what the downstream RL curriculum builds on.

How to use

# In the LiteResearcher training scripts (Training/ folder of the repo)
export MODEL_PATH=$(hf download simplex-ai-inc/LiteResearcher-4B-SFT \
                                --local-dir ./literesearcher_sft)

Then follow the Stage-1 / Stage-2 RL instructions in the LiteResearcher repository.

Stand-alone inference

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_id = "simplex-ai-inc/LiteResearcher-4B-SFT"
tok = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id, torch_dtype=torch.bfloat16, device_map="auto"
)

The model expects the same ReAct system prompt and tool schema used by LiteResearcher (see Inference/ in the repo).

Citation

If you use this checkpoint in academic work, please cite the LiteResearcher project β€” see the GitHub README for the BibTeX entry.

License

Apache-2.0, inheriting from the Qwen3-4B-Thinking-2507 base model.

agent
cold-start
conversational
deep-research
endpoints_compatible
LiteResearcher
qwen3
react
safetensors
search
text-generation
text-generation-inference
thinking
transformers

simplex-ai-inc/LiteResearcher-4B-SFT

Model

LiteResearcher-4B-SFT

0

4 commits

3 linked in READMEs

updated Jun 9, 2026

See the code

README

LiteResearcher-4B-SFT

This is the SFT cold-start checkpoint for LiteResearcher β€” a scalable agentic RL training framework for deep-research agents.

It is the initial policy used to launch the two-stage curriculum RL training that produces the final simplex-ai-inc/LiteResearcher-4B model.

If you are looking for the final RL model, please use simplex-ai-inc/LiteResearcher-4B. If you want to reproduce the RL training from scratch, this is the checkpoint you need.

Model details

  • Base model: Qwen/Qwen3-4B-Thinking-2507
  • Architecture: Qwen3ForCausalLM (36 layers, hidden 2560, 32 heads, GQA 8 KV heads)
  • Max position embeddings: 262,144 (RoPE ΞΈ = 5,000,000)
  • Precision: bfloat16
  • Total params: ~4B
  • Training framework: LLaMA-Factory

Training recipe

ItemValue
StageSFT (cold-start before RL)
Base modelQwen/Qwen3-4B-Thinking-2507
Datasetsimplex-ai-inc/LiteResearcher-Data (~68.2k SFT trajectories)
Max sequence length64K (cutoff_len=65536)
Global batch size128 (per-device bs 2 Γ— grad-accum 8 Γ— 8 GPUs)
Epochs1
Optimizer steps533
Learning rate2.0e-5, cosine, 10% warmup
Final train lossβ‰ˆ 0.447 (starting loss β‰ˆ 1.19)

The SFT trajectories teach the model the ReAct think β†’ search β†’ visit β†’ answer loop and the strict <answer>...</answer> output contract used by the RL environment. Because the base is the Thinking-2507 variant, the model preserves long chain-of-thought behavior inside <think>...</think> blocks, which is what the downstream RL curriculum builds on.

How to use

# In the LiteResearcher training scripts (Training/ folder of the repo)
export MODEL_PATH=$(hf download simplex-ai-inc/LiteResearcher-4B-SFT \
                                --local-dir ./literesearcher_sft)

Then follow the Stage-1 / Stage-2 RL instructions in the LiteResearcher repository.

Stand-alone inference

from transformers import AutoModelForCausalLM, AutoTokenizer
import torch

model_id = "simplex-ai-inc/LiteResearcher-4B-SFT"
tok = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(
    model_id, torch_dtype=torch.bfloat16, device_map="auto"
)

The model expects the same ReAct system prompt and tool schema used by LiteResearcher (see Inference/ in the repo).

Citation

If you use this checkpoint in academic work, please cite the LiteResearcher project β€” see the GitHub README for the BibTeX entry.

License

Apache-2.0, inheriting from the Qwen3-4B-Thinking-2507 base model.

agent
cold-start
conversational
deep-research
endpoints_compatible
LiteResearcher
qwen3
react
safetensors
search
text-generation
text-generation-inference
thinking
transformers