Staff ML Engineer: End-to-End Production AI Systems

Bjak; Indonesia; On-site; IDR 450,000,000 - 700,000,000; Full time

Job summary

Bjak is seeking a Staff Machine Learning Engineer to own the end-to-end ML systems that power our products. You will lead data, training, evaluation, inference, and deployment, turning research into production-ready solutions.

You will design high-performance pipelines, build scalable training and fine-tuning workflows, and collaborate with research and engineering to ship reliable AI features at scale.

Qualifications

• Experience building and shipping ML systems in production.
• Strong understanding of large-model training, fine-tuning, evaluation, and inference.
• Strong software engineering and systems fundamentals.
• Experience operating ML workloads at scale, particularly GPU-based systems.
• Strong technical judgment and the ability to work independently.
• Bias toward experimentation, measurement, and shipping.
• High standards for correctness and production quality.

Responsibilities

• Own the end-to-end ML systems powering our company, from data and training to evaluation, inference, and deployment.
• Build and evolve training and fine-tuning pipelines for large models.
• Design evaluation systems that measure capability, robustness, safety, and real-world product performance.
• Architect high-performance inference systems, optimizing latency, GPU utilization, memory, cost, and reliability.
• Build data pipelines for high-quality real-world and synthetic training data.
• Establish production infrastructure for deploying, monitoring, and improving models.
• Partner with research and application engineering to turn model capabilities into product improvements.
• Make pragmatic technical trade-offs and iterate based on real-world performance.

Skills

Production ML systems

GPU-based workloads

Python

Software engineering

Independent problem solving

Tools

PyTorch

JAX

Job description

Bjak is seeking a Staff Machine Learning Engineer to own the end-to-end ML systems that power our products. You will lead data, training, evaluation, inference, and deployment, turning research into production-ready solutions.

You will design high-performance pipelines, build scalable training and fine-tuning workflows, and collaborate with research and engineering to ship reliable AI features at scale.