AI Engineer (Legal Domain LLM), Traineeship, [GRIT@Gov]

Careers@Gov; Singapore; On-site; SGD 80,000 - 100,000; Full time

Job summary

A government agency in Singapore is seeking a motivated fresh graduate to join its AI engineering team focused on developing solutions using large language models (LLMs). You will collaborate with product managers and researchers to grow a legal domain model, gaining hands-on experience in machine learning techniques. Ideal candidates should hold a relevant degree and have basic coding skills in Python, alongside a keen interest in Natural Language Processing and AI task optimization.

Qualifications

• Degree holder in relevant engineering discipline or law with technology modules.
• Basic understanding of machine learning fundamentals, including deep learning architectures.

Responsibilities

• Engage lawyer volunteers and analyse demand use cases.
• Contribute to training and evaluation of large language models.
• Assist in developing data processing pipelines.

Skills

Python coding experience

Web programming

Interest in Natural Language Processing

Good communication skills

Ability to thrive in a fast-paced environment

Education

Degree in Computer Science, Engineering, Information Technology, or relevant discipline

Job description

[What the role is]

One of IMDA BizTech Group’s interest area is to develop solutions based on large language models (LLMs) that drive real‑world innovation. Our team includes AI scientists and researchers, data engineers, and platform engineers who are passionate about scalable AI systems and ethical responsible AI development.

We’re seeking a motivated and curious fresh graduate to join our AI engineering team. You will work closely with senior engineers to design, train, and evaluate large language models (LLMs). This is a hands‑on role where you’ll gain deep exposure to cutting‑edge machine learning techniques and infrastructure at scale.

[What you will be working on]

You will be working alongside product managers, AI scientists and engineers as one team to grow a legal domain model, GPT‑Legal. Responsibilities include :

• Engage lawyer volunteers and analyse demand use cases
• Interpret the use cases and translate to specific model tasks for AI scientist to develop model training strategies
• Hands‑on with lawyers to create and validate training data and model performance
• Contribute to the training, fine‑tuning, and evaluation of LLMs.
• Assist in developing data processing pipelines and model training workflows
• Collaborate with researchers on algorithmic improvements and performance optimization.
• Test and deploy models using distributed computing tools.
• Document experiments, results, and processes systematically.

[What we are looking for]

Degree holder in Computer Science, Engineering, Information Technology, or in relevant engineering discipline or law with technology modules

Basic understanding of machine learning fundamentals, including deep learning architectures

Coding experience in Python and Web programming

Interest or prior project in either one of the following areas :

• Natural Language Processing (NLP), recommendation system and / or Synthetic AI
• AI tasks optimisation

Good communication skills, both verbal and written

Ability to thrive in a dynamic, fast‑paced environment

Note : Only shortlisted candidates will be contacted