About this role
AI Singapore’s AI Products Pillar seeks an AI Engineer to develop, train, evaluate and engineer multilingual, multicultural and multimodal foundation large language models, focusing on underrepresented languages and cultures and real-world applications. The role is hosted at Nanyang Technological University’s NTU Main Campus in Singapore under the VP (Artificial Intelligence & Digital Economy)’s office.
Responsibilities
- Develop and adapt LLMs through continued pre-training, supervised fine-tuning, instruction tuning and preference optimisation; evaluate model performance.
- Build scalable multilingual text pipelines for ingestion, cleaning, filtering, deduplication, tokenisation, contamination detection and dataset mixture creation.
- Support distributed training through experiment tracking, checkpointing, monitoring, debugging, profiling and compute optimisation.
- Review and reproduce frontier AI research, turning findings into improvements in training, evaluation and deployment. Produce reusable scripts, evaluation tools, model artefacts, APIs, documentation and deployment-ready components.
- Collaborate with internal teams, research labs, academic institutions and industry partners. Contribute to technical sharing, workshops, publications and other outreach where relevant.
Requirements
- Proficiency in Python and sound software engineering practices, including modular code, Git, testing, documentation and code review. Hands-on experience with PyTorch and NLP/LLM frameworks, particularly Hugging Face Transformers, Datasets, Tokenizers, Accelerate and TRL.
- Experience or strong working knowledge of LLM training, tuning, evaluation and adaptation; familiarity with low-resource languages, code-switching, language imbalance, script diversity, tokenisation and multilingual benchmarks.
- Ability to build scalable text, dataset and training pipelines; familiarity with distributed and GPU-based training, training stability, checkpointing, experiment tracking and compute optimisation. Familiarity with Linux, Docker, CI/CD, cloud platforms or GPU/HPC clusters.
- Ability to assess research papers, design experiments and ablation studies, analyse results and communicate technical trade-offs to varied stakeholders. The team values self-direction, collaboration, reproducibility, reliability and responsible AI development.
Applicants from all disciplines and qualification pathways, including NITEC, Diploma, Bachelor’s, Master’s, PhD or equivalent practical experience, are welcome. A formal qualification in a related technical field is advantageous but not mandatory. Experience with reproducible model workflows, model versioning, automated evaluation and API deployment or serving tools such as vLLM, SGLang or TGI is advantageous.
For full details about this role, click “Apply on company site” to view the employer’s original posting.