Lead ML Engineer: Build Production-Grade AI Systems

ActAI; Indonesia; On-site; IDR 360,000,000 - 480,000,000; Full time

Job summary

ActAI is seeking a Lead Engineer, Machine Learning to own end-to-end ML systems—from data and training to deployment and evaluation. You will design scalable training pipelines, optimize inference latency, and drive production-grade reliability across models and systems.

Ideal candidates ship real ML systems, operate at scale on GPU hardware, and bridge research with engineering. A hands-on leader who ships with measurable impact is essential for our small, fast-moving team.

Qualifications

• Experience building and shipping ML systems used in production.
• Strong understanding of modern large-model training, fine-tuning, evaluation, and inference.
• Strong software engineering and systems fundamentals.
• Experience operating ML workloads at meaningful scale, particularly GPU-based systems.
• Strong technical judgment and the ability to navigate ambiguous problems independently.
• A bias toward experimentation, measurement, and shipping.
• High standards for correctness, reliability, 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 and systems for high-quality real-world and synthetic training data.
• Establish reliable production infrastructure for deploying, monitoring, and continuously improving models.
• Partner closely with research and application engineering to turn model capabilities into product improvements.
• Make pragmatic technical trade-offs and rapidly iterate based on real-world performance.

Skills

ML systems

GPU training

Python

Distributed systems

Tools

PyTorch

JAX

CUDA

Job description

ActAI is seeking a Lead Engineer, Machine Learning to own end-to-end ML systems—from data and training to deployment and evaluation. You will design scalable training pipelines, optimize inference latency, and drive production-grade reliability across models and systems.

Ideal candidates ship real ML systems, operate at scale on GPU hardware, and bridge research with engineering. A hands-on leader who ships with measurable impact is essential for our small, fast-moving team.