Lead AI Engineer, AI Products

Job Responsibilities

• Design, develop, integrate and optimise scalable AI/ML solutions, including LLM-based applications, Retrieval-Augmented Generation (RAG), agentic AI and multimodal AI solutions to address Home Team use cases.
• Handle data pre-processing, engineering and annotation for model training and development.
• Work closely with the software engineers to write production-quality code and deploy AI products powered by AI/ML, RAG and agentic AI for use by Home Team on Cloud Platform (GCC) or on-prem infrastructure.
• Design and implement model evaluation, monitoring and optimise performance of in-house developed AI solutions through pre-training and post-training techniques.
• Optimise model inference performance and resource utilisation.
• Develop and deploy risk mitigation measures such as guardrails to ensure safe and responsible use of AI products, aligning with organisational standards and regulatory requirements.
• Work closely with Product Management team to translate product requirements into machine learning / deep learning model requirements.
• Provide technical advisory and review related technical articles to support key business decision and project decision making.
• Troubleshoot complex production AI issues and implement improvements to enhance system reliability and operational performance.
• Keep abreast of new AI technologies development and continuously improve HTxAI’s AI engineering capability.
• Learn and share knowledge in a multi-disciplinary team both internally and externally.

Requirements

Job Requirements

• Postgraduate / Tertiary qualification in any of the following fields: Computer Science, Computer Engineering, Data Science, Data Analytics, Machine Learning, or any relevant areas.
• Min. 6 years of software engineering and AI engineering experience, including at least 3 years developing and deploying production AI applications.
• Strong proficiency in Python with extensive experience developing production-quality software.
• Strong hands-on experience designing, developing, deploying and maintaining production AI/ML solutions.
• Strong experience with open-source AI frameworks such as LangChain / LangGraph, and open models such as Llama 4 and GPT OSS.
• Extensive experience in building scalable AI applications and deploy applications using Docker containers with REST APIs.
• Experience deploying AI application using Kubernetes.
• Experience with vector databases and model serving frameworks.
• Able to work collaboratively with cross-functional teams multi-disciplinary / multi-agencies teams.
• Able to communicate technical concepts and conduct demos in a clear and effective manner to stakeholders of varying technical backgrounds.
• Knowledge of emerging AI-related technologies and trends.
• Self-motivated, team player, creative with excellent communication skills, excellent interpersonal skills and ability to work independently.
• Having domain knowledge in Homeland Security sector will be an added advantage.

What We Offer

• Opportunity to work on meaningful public sector digital initiatives.
• A supportive, purpose-driven team culture.
• Structured career development and technical upskilling.
• Access to modern tools, cloud technologies, and secure DevSecOps environments.

Additional Information

• Shortlisted applicants may be involved in classified or sensitive projects that require handling of restricted information. As such, selected candidates will be required to undergo a security clearance process as part of the employment consideration.