Technical Lead, Machine Learning

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

Own the execution layer of a proactive AI assistant by translating research into reliable, scalable, production-grade ML systems. Build end-to-end pipelines—training workflows, evaluation, inference architecture, and deployment—while fine-tuning with methods like LoRA/QLoRA, SFT, DPO, and distillation. Optimize GPU performance for low-latency, high-reliability inference, and partner with research and application engineering to integrate models across backend, mobile, and desktop products.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
BJAK
BJAK
1 day ago

Technical Lead, Machine Learning

✓ 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 a proactive AI assistant by translating research into reliable, scalable, production-grade ML systems. Build end-to-end pipelines—training workflows, evaluation, inference architecture, and deployment—while fine-tuning with methods like LoRA/QLoRA, SFT, DPO, and distillation. Optimize GPU performance for low-latency, high-reliability inference, and partner with research and application engineering to integrate models across backend, mobile, and desktop products.
Location: Beijing
Workplace: Remote
Employment Type: Full time
Job Function: Data Science & Machine Learning

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 training data systems for synthetic and real-world data.
  • •Implement evaluation pipelines for performance, robustness, safety, and bias, and own production deployment including GPU optimization, memory efficiency, latency reduction, and scaling policies.

Key Requirements

  • •Have built or shipped real ML systems used by people, not just demos.
  • •Comfortable working with large models and understanding their failure modes.
  • •Write strong, production-grade code with a focus on system correctness.
  • •Self-directed and pragmatic, taking full ownership of outcomes.
  • •Communicate clearly and collaborate well in small, high-trust teams.
Experience:Machine learning
Skills:OwnershipSelf-directedPragmatismCommunicationCollaboration
Tech Stack:PythonPyTorchJAXGPU-based trainingGPU-based inference

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