BJAK · 19 hours ago
Staff Machine Learning Engineer
United States
Full-time
Remote
Lead/StaffBJAK is building a proactive AI system that understands context across conversations and plans actions. The Staff Machine Learning Engineer will be responsible for turning research direction into production-grade ML systems, owning the execution layer of A1’s intelligence.
Information TechnologyInsurTechPrice Comparison
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
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
Qualification
Required
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
Python
PyTorch / JAX
GPU-based training and inference system
You have built or shipped real ML systems used by people, not just demos
You are comfortable working with large models and understanding their failure modes
You write strong, production-grade code and care about system correctness
You are self-directed, pragmatic, and take full ownership of outcomes
You communicate clearly and collaborate well in small, high-trust teams
Company
BJAK
Bjak is a technology company that develops an insurance comparison platform.
Founded in 2019
201-500 employees
Funding
Current Stage
Growth StageLeadership Team
Kuan Yue Ronald Lim
Executive Director
Recent News
Fintech News Malaysia
2025-12-16
2025-12-16
2025-11-29
Company data provided by crunchbase