Senior AI & Machine Learning Engineer

SGInnovate; Singapore; On-site; SGD 150,000 - 230,000; Full time

Benefits offered by this job

Health insurance

Learning budget

Job summary

HeyHi, Singapore's EdTech leader, seeks a Senior AI & Machine Learning Engineer to design and deploy scalable AI-driven learning systems. You will mentor junior engineers, shape technical direction, and collaborate with product, engineering, and leadership to advance adaptive learning platforms.

You will work on end-to-end AI pipelines, including data ingestion, model training, evaluation, deployment, and monitoring, with a strong emphasis on responsible AI and scalability across K–12, higher

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Data Science, AI, or a related field.
• 5+ years hands-on experience in AI/ML systems, or data engineering, with a proven track record of shipping AI/ML products to production.
• Proficiency in Python and ML frameworks such as TensorFlow, PyTorch, and Hugging Face.
• Experience fine-tuning and evaluating LLMs/SLMs on domain-specific data, including RAG, prompt engineering, and model benchmarking.
• Experience with agentic AI frameworks and patterns (e.g., LangGraph, LangChain, LlamaIndex, AutoGen, CrewAI) including multi-agent orchestration, tool use, planning, and expert routing.
• Hands-on experience designing and deploying knowledge graphs and graph databases, including ontology modelling and schema design for domain data.
• Experience with multimodal AI — speech and/or vision — is highly desirable.
• Proven ability to mentor engineers and lead technical initiatives from concept to delivery.
• Excellent problem-solving, communication, and collaboration skills in a cross-functional environment.
• Passion for education and a genuine interest in using AI to improve learning outcomes.

Responsibilities

• Lead the design, development, and deployment of scalable, production-grade AI/ML models and systems for educational applications.
• Architect end-to-end AI/ML pipelines including data ingestion, feature engineering, model training, evaluation, deployment, and monitoring (MLOps).
• Drive research and experimentation with LLMs, Agentic AI, NLP, recommendation systems, and adaptive learning algorithms.
• Partner with product, engineering, and business stakeholders to translate complex user needs into intelligent, measurable solutions.
• Mentor and provide technical guidance to junior AI/ML engineers, fostering a culture of engineering excellence.
• Define best practices for model development, code quality, testing, reproducibility, and responsible AI.
• Own the performance, reliability, and scalability of deployed solutions, including continuous monitoring, retraining, and optimisation.
• Evaluate and introduce emerging AI tools, frameworks, and techniques to keep HeyHi at the cutting edge of EdTech innovation.
• Contribute to technical roadmaps, architectural decisions, and long-term AI strategy.

Skills

AI/ML systems

Python

ML frameworks

Education

Bachelor’s/Master’s in CS/Data Science/AI

Tools

TensorFlow

PyTorch

HuggingFace

LangChain

LangGraph

LlamaIndex

AutoGen

CrewAI

Job description

Background

HeyHi is reshaping education with AI! We’re an EdTech company on a mission to transform teaching and learning with our AI-powered E-Assessment and Personalised Adaptive Learning System. We empower educational institutions across K–12 and higher education, as well as enterprises, by transforming static learning materials into adaptive nano-learning content such as flashcards, e-books, videos, and quizzes, combined with personalised assessments. HeyHi serves educators as a co-teacher or co-instructor, supporting lesson delivery, content creation, and learner engagement. Powered by AI-generated knowledge points and knowledge graphs grounded in a verified knowledge base, HeyHi delivers dynamic learning pathways tailored to each learner’s needs, making learning more efficient, personalised, and transparent. HeyHi is a GESAward 2024 and 2025 winner, an Institute of Adult Learning InnovSpur Challenge 2025 grant awardee, and an Enterprise Singapore Startup SG Tech grant awardee.

HeyHi is seeking a Senior AI & Machine Learning Engineer to join our Singapore team. In this senior role, you will lead the design, architecture, and deployment of intelligent systems that personalise learning experiences at scale. You will mentor junior engineers, set technical direction for AI and machine learning initiatives, and work closely with product, engineering, and leadership to shape the future of our EdTech platform. This is an exciting opportunity to drive meaningful innovation across academic and professional learning environments.

Job scope

Key Responsibilities:

• Lead the design, development, and deployment of scalable, production-grade AI/ML models and systems for educational applications.
• Architect end-to-end AI/ML pipelines including data ingestion, feature engineering, model training, evaluation, deployment, and monitoring (MLOps).
• Drive research and experimentation with LLMs, Agentic AI, NLP, recommendation systems, and adaptive learning algorithms.
• Partner with product, engineering, and business stakeholders to translate complex user needs into intelligent, measurable solutions.
• Mentor and provide technical guidance to junior AI/ML engineers, fostering a culture of engineering excellence.
• Define best practices for model development, code quality, testing, reproducibility, and responsible AI.
• Own the performance, reliability, and scalability of deployed solutions, including continuous monitoring, retraining, and optimisation.
• Evaluate and introduce emerging AI tools, frameworks, and techniques to keep HeyHi at the cutting edge of EdTech innovation.
• Contribute to technical roadmaps, architectural decisions, and long‑term AI strategy.

Requirements:

• Bachelor’s or Master’s degree in Computer Science, Data Science, AI, or a related field.
• 5+ years of hands‑on experience in AI/ML systems, or data engineering, with a proven track record of shipping AI/ML products to production.
• Deep proficiency in Python and strong experience with ML frameworks such as TensorFlow, PyTorch, and Hugging Face.
• Proven experience fine‑tuning and evaluating LLMs/SLMs on domain‑specific data, including RAG, prompt engineering, and model benchmarking.
• Strong experience with agentic AI frameworks and patterns (e.g., LangGraph, LangChain, LlamaIndex, AutoGen, CrewAI) including multi‑agent orchestration, tool use, planning, and expert routing.
• Hands‑on experience designing and deploying knowledge graphs and graph databases, including ontology modelling and schema design for domain data.
• Experience with multimodal AI — speech (ASR, prosody) and/or vision (OCR, handwriting, diagram) — is highly desirable.
• Proven ability to mentor engineers and lead technical initiatives from concept to delivery.
• Excellent problem‑solving, communication, and collaboration skills in a cross‑functional environment.
• Passion for education and a genuine interest in using AI to improve learning outcomes.