Principal Machine Learning Engineer
Avomind Open · verified Oct 22, 2026 At a glance
Senior Machine Learning Engineering role at Avomind. Remote, Singapore · full-time.
Pay not stated Machine Learning Engineer salary
aimlroles summary, based on the employer's posting.
What you'll do
- Build the full ML workflow, from preparing data and training models through evaluation, inference and deployment
- Adapt models with techniques such as LoRA, QLoRA, supervised fine-tuning, preference optimization and distillation
- Run inference systems at scale, balancing response speed, operating cost and reliability
- Create data pipelines for synthetic and real-world training material, plus evaluations of model quality, safety and bias
- Improve production performance through GPU and memory tuning, then connect ML systems to backend, mobile and desktop apps
What you bring
- You have deep learning expertise, including transformer architectures, and have trained, adapted or deployed large models in production
- You can build maintainable production software and independently take ML systems from start to finish
- You work with PyTorch or JAX and have experience with distributed training or inference tools such as DeepSpeed, FSDP or another listed framework
- You have optimized GPU workloads, including memory use, quantization or mixed-precision processing
Who this fits
This senior role suits an engineer who can take ownership of production ML systems and work closely with research and application engineering teams. The role is remote in Singapore.
From the employer
About the Company
Our client is a stealth AI startup backed by one of Southeast Asia's leading technology companies and is currently building its global founding team.
The company is developing an AI-native communication platform designed to simplify everyday tasks by integrating AI directly into conversations. Instead of switching between multiple applications, users can plan, organize, compare, research, and complete tasks within a single intelligent assistant.
Serving a market of billions of users still relying on traditional productivity tools, the platform focuses on delivering reliable AI workflows, persistent context, multi-step reasoning, and seamless task execution. The mission is to create an AI assistant that significantly improves productivity while making everyday work simpler and more intuitive.
About the Role
Our client is seeking a Principal Machine Learning Engineer to build and deploy production-grade machine learning systems that power its AI platform. This role focuses on translating research into scalable solutions by developing robust training pipelines, inference systems, evaluation frameworks, and deployment infrastructure.
Working closely with research and application engineering teams, this position will play a key role in delivering reliable, high-performance ML systems that operate effectively under real-world production constraints.
Key Responsibilities
- Build and own end-to-end machine learning pipelines covering data processing, model training, evaluation, inference, and deployment.
- Fine-tune and adapt models using modern techniques such as LoRA, QLoRA, Supervised Fine-Tuning (SFT), Direct Preference Optimization (DPO), and model distillation.
- Design and operate scalable inference systems while balancing latency, cost, and reliability.
- Develop and maintain data pipelines for both synthetic and real-world training datasets.
- Build evaluation frameworks to assess model performance, robustness, safety, and bias in collaboration with research teams.
- Optimize production deployments through GPU optimization, memory efficiency, latency reduction, and scaling strategies.
- Collaborate with application engineering teams to integrate machine learning systems into backend, mobile, and desktop applications.
- Continuously improve ML systems through rapid iteration and real-world performance monitoring while balancing production constraints such as latency, cost, reliability, and safety.
Requirements
- Strong background in deep learning and transformer-based architectures.
- Hands-on experience training, fine-tuning, or deploying large-scale machine learning models in production.
- Proficiency with modern machine learning frameworks such as PyTorch or JAX.
- Experience with distributed training and inference frameworks, including technologies such as DeepSpeed, FSDP, Megatron, ZeRO, or Ray.
- Strong software engineering skills with experience building robust, maintainable, production-grade systems.
- Experience optimizing GPU workloads, including memory efficiency, quantization, and mixed precision.
- Ability to independently own end-to-end machine learning systems in fast-moving environments.
- Strong problem-solving skills with a focus on rapid iteration and continuous improvement.
Preferred Qualifications
Experience with one or more of the following is preferred:
- LLM inference frameworks such as vLLM, TensorRT-LLM, or FasterTransformer
- Open-source contributions to machine learning or systems libraries
- Scientific computing, compiler technologies, or GPU kernel development
- Reinforcement Learning from Human Feedback (RLHF) pipelines, including PPO, DPO, or ORPO
- Training or deploying multimodal or diffusion models
- Large-scale data processing frameworks such as Apache Arrow, Spark, or Ray