Staff Machine Learning Engineer

BJAK
Beijing
Workplace: RemoteFull timeFunction: Data Science & Machine LearningSkills: ["Communication","Collaboration","Ownership","Pragmatism","Judgment"]

Own the execution layer of an AI assistant by translating research into reliable, scalable, production-grade ML systems. Build end-to-end ML pipelines—data, training, evaluation, inference architecture, and deployment—using modern fine-tuning and distillation techniques. Architect GPU-optimized inference to balance latency, cost, and reliability, and define evaluation for performance, robustness, safety, and bias while partnering with application engineering to integrate into backend and desktop/mobile products.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
BJAK
BJAK
1 day ago

Staff Machine Learning Engineer

✓ Verified Job

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

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

Job Summary

Own the execution layer of an AI assistant by translating research into reliable, scalable, production-grade ML systems. Build end-to-end ML pipelines—data, training, evaluation, inference architecture, and deployment—using modern fine-tuning and distillation techniques. Architect GPU-optimized inference to balance latency, cost, and reliability, and define evaluation for performance, robustness, safety, and bias while partnering with application engineering to integrate into backend and desktop/mobile products.
Location: Beijing
Workplace: Remote
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Own end-to-end ML system execution including data pipelines, training workflows, evaluation systems, inference architecture, and deployment.
  • •Fine-tune and adapt models using 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 for performance, robustness, safety, and bias and ensure reliable production deployment with GPU optimization and scaling policies.

Key Requirements

  • •Built or shipped real ML systems used by people, not just demos.
  • •Comfortable working with large models and understanding their failure modes.
  • •Writes production-grade code with a strong focus on system correctness.
  • •Self-directed and pragmatic, taking full ownership of outcomes.
  • •Communicates clearly and collaborates well in small, high-trust teams.
Experience:Machine learningLarge models
Skills:CommunicationCollaborationOwnershipPragmatismJudgment
Tech Stack:PythonPyTorchJAXLoRAQLoRASFTDPODistillationGPUTrainingInference

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