Home
Jobs
Saved
Resumes
Principal Machine Learning Engineer at BJAK | JobVerse
JobVerse
Home
Jobs
Recruiters
Companies
Pricing
Blog
Jobs
/
Principal Machine Learning Engineer
BJAK
Remote
Website
LinkedIn
Principal Machine Learning Engineer
Singapore
Full Time
2 weeks ago
No Sponsorship
Apply Now
Key skills
PyTorch
Ray
ML
Deep Learning
JAX
Leadership
About this role
Role Overview
Build and own end-to-end ML pipelines spanning data, training, evaluation, inference, and deployment.
Fine-tune and adapt models using state-of-the-art methods such as LoRA, QLoRA, SFT, DPO, and distillation.
Architect and operate scalable inference systems, balancing latency, cost, and reliability.
Design and maintain data systems for high-quality synthetic and real-world training data.
Implement evaluation pipelines covering performance, robustness, safety, and bias, in partnership with research leadership.
Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies.
Collaborate closely with application engineering to integrate ML systems cleanly into backend, mobile, and desktop products.
Make pragmatic trade-offs and ship improvements quickly, learning from real usage.
Work under real production constraints: 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 ML models in production.
Proficiency with at least one modern ML framework (e.g. PyTorch, JAX), and ability to learn others quickly.
Experience with distributed training and inference frameworks (e.g. DeepSpeed, FSDP, Megatron, ZeRO, Ray).
Strong software engineering fundamentals – you write robust, maintainable, production-grade systems.
Experience with GPU optimization, including memory efficiency, quantization, and mixed precision.
Comfort owning ambiguous, zero-to-one ML systems end-to-end.
A bias toward shipping, learning fast, and improving systems through iteration.
Tech Stack
PyTorch
Ray
Benefits
Health insurance
Flexible work arrangements
Professional development opportunities
Apply Now
Home
Jobs
Saved
Resumes