Principal Machine Learning Engineer

Build the next generation of AI assistants.A1 is an AI lab focused on building a proactive AI assistant that helps users manage conversations, automate workflows, organize tasks, and complete real-world actions with intelligence and reliability.We're looking for an experienced Machine Learning Engineer to design and build production-grade AI systems that power long-running workflows, persistent memory, reasoning, and reliable task execution. This is an opportunity to work on cutting-edge LLM technologies and help shape the future of AI-powered products.Key ResponsibilitiesDesign, build and own end-to-end machine learning pipelines covering data preparation, training, evaluation, inference and deployment.Fine-tune and optimize large language models using modern techniques such as LoRA, QLoRA, Supervised Fine-Tuning (SFT), Direct Preference Optimization (DPO) and model distillation.Build and maintain scalable inference infrastructure with a focus on latency, reliability, GPU utilization and cost optimization.Develop high-quality synthetic and real-world data pipelines to support model training and continuous improvement.Design automated evaluation frameworks to measure model quality, robustness, safety and bias.Optimize GPU performance through quantization, mixed precision and memory-efficient inference techniques.Deploy and operate production AI systems that serve real users at scale.Collaborate closely with software engineers and product teams to integrate ML capabilities across backend, mobile and desktop applications.Continuously improve system performance through experimentation, monitoring and iterative development.RequirementsBachelor's, Master's or PhD in Computer Science, Machine Learning, Artificial Intelligence or a related discipline.Strong understanding of deep learning fundamentals and Transformer architectures.Hands-on experience training, fine-tuning or deploying large language models in production environments.Strong proficiency in PyTorch, JAX or equivalent machine learning frameworks.Experience with distributed training and inference technologies such as DeepSpeed, FSDP, Megatron-LM, ZeRO or Ray.Solid software engineering skills with the ability to develop scalable, maintainable production systems.Experience optimizing GPU performance, memory efficiency, mixed precision and quantization.Comfortable working in fast-paced, ambiguous, zero-to-one environments with a strong ownership mindset.Passion for building high-quality AI products that solve real-world problems.Preferred QualificationsExperience with one or more of the following is highly advantageous:vLLM, TensorRT-LLM or FasterTransformerRLHF techniques such as PPO, DPO or ORPOMultimodal or diffusion modelsLarge-scale data processing using Spark, Apache Arrow or RayGPU kernel optimization or scientific computingContributions to open-source AI or machine learning projectsWhy Join A1?Build production AI systems using the latest LLM technologies.Work on challenging problems involving reasoning, memory, planning and autonomous AI agents.Join a high-performing, engineering-driven team where you'll have meaningful ownership and direct impact.Collaborate with experienced AI researchers and engineers to bring cutting-edge research into production.Help shape products that aim to redefine how people interact with AI every day.How To Apply:Submit your application by emailing a detailed copy of your updated Resume in MS Word Format to Lin Jin Sen (EA Personnel Reg. No. R25129852, Achieve Career Consultant Pte Ltd EA Licence No. 05C3451)by clicking the “Apply Now” or email to [email protected] for a confidential discussionCurrent & Expected SalaryReason(s) for leavingNotice Period / Availability to commence workYOUR SUCCESS IS OUR ACHIEVEMENT!By submitting any application or résumé to us, you will be deemed to have agreed and consented to us collecting, using, retaining and disclosing your personal information to prospective employers for their consideration.Cessation Of Collection Of Full NRIC Numbers:In compliance with the Personal Data Protection Act and commitment to protect candidates’ personal data, Achieve Group will cease to collect, process or use full NRIC numbers during our screening and job application process.Kindly ensure your resumes provided to us does not contain your full NRIC number and full home address during your job application.We regret that only shortlisted candidates will be notified.