Principal Machine Learning Engineer

BJAK
Beijing
Workplace: RemoteFull timeFunction: Data Science & Machine LearningSkills: ["PyTorch","JAX","LLM","Distributed training","GPU optimization","Quantization","Mixed precision","Production-grade systems","Software engineering"]

Lead the end-to-end development of production-grade ML systems, from data pipelines and training to deployment and inference. You’ll optimize GPU usage, fine-tune models with LoRA/QLoRA/SFT/DPO, and collaborate with engineering teams to ship scalable AI features under real production constraints.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
BJAK
BJAK
6 months ago

Principal Machine Learning Engineer

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 18 hours agoStatus: Live

Job Summary

Lead the end-to-end development of production-grade ML systems, from data pipelines and training to deployment and inference. You’ll optimize GPU usage, fine-tune models with LoRA/QLoRA/SFT/DPO, and collaborate with engineering teams to ship scalable AI features under real production constraints.
Location: Beijing
Workplace: Remote
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Sr. Manager level

Key Responsibilities

  • •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.

Key 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:AI
Skills:PyTorchJAXLLMDistributed trainingGPU optimizationQuantizationMixed precisionProduction-grade systemsSoftware engineering
Tech Stack:PyTorchJAXDeepSpeedFSDPMegatronZeRORayVLLMTensorRT-LLMFasterTransformerApache ArrowSpark

Company Brief

BJAK
Bjak is a Malaysia-based insurtech that operates an online insurance comparison and distribution platform across SEA, simplifying purchase and servicing of motor and life insurance using digital tools and AI-enabled services.
Industry: InsurTech
Company Size: Medium (51 to 250 employees)
Growth: Growth Stage Startup
Headquarters: Selangor, Malaysia
Founded: 2019
Glassdoor
Glassdoor: 2.5
WebsiteLinkedInGlassdoor