Lead AI R&D Engineer: LLM Post-Training & Agentic Tools

Home Team Science and Technology Agency (HTX); Singapore; On-site; SGD 120,000 - 180,000; Full time

Job summary

Home Team Science and Technology Agency (HTX) seeks a skilled Applied AI Scientist to post-train sovereign LLMs for agentic workflows. You will shape RL environments, harnesses, and checkpoints, enabling safe, multi-step task execution and tool usage within our GPU clusters.

You will work on end-to-end training loops, data curation, and model fine-tuning while collaborating with HTxAI projects to advance state-of-the-art sovereign AI capabilities for Singapore's Home Team.

Qualifications

• Bachelor's degree in CS/DS/Engineering/Math or equivalent experience.
• Hands-on experience fine-tuning or training LLMs in PyTorch.
• Strong Python and Linux proficiency.
• Willingness to learn and experiment in a fast-paced Applied AI R&D environment.

Responsibilities

• Post-train LLMs for agentic workflows, including RL environments and training tasks.
• Own the loop in a selected post-training domain, including environments, data, rewards and fine-tuning runs.
• Build and scale tool-use environments and harnesses for agentic RL training.
• Maintain containerised sandboxes and ensure security, access controls and testing.

Skills

Python

PyTorch

LLM fine-tuning

Linux

Reinforcement learning

GPU clusters

Education

Bachelor's degree in Computer Science, Data Science, Engineering, Mathematics or related field

Job description

Home Team Science and Technology Agency (HTX) seeks a skilled Applied AI Scientist to post-train sovereign LLMs for agentic workflows. You will shape RL environments, harnesses, and checkpoints, enabling safe, multi-step task execution and tool usage within our GPU clusters.

You will work on end-to-end training loops, data curation, and model fine-tuning while collaborating with HTxAI projects to advance state-of-the-art sovereign AI capabilities for Singapore's Home Team.