Latest AI JobsBrowse Jobs
bjakcareer logo

Lead Engineer, Machine Learning

bjakcareerIndonesiaActAI Engineering
Apply on company site
Salary
Not listed
Job type
Full-time
Work mode
Remote
Experience
Lead
Posted
28 Sept 2026
Status
Open · checked 7d ago

About this role

Based in Indonesia, the Lead Engineer, Machine Learning position at bjakcareer involves directing the execution layer of intelligence to transition models into production systems. This hands-on leadership role suits an experienced machine learning professional who wants to operate at the intersection of research, systems, and product development.

Role Overview

The organization focuses on creating proactive applications designed to assist users with conversations, errands, organizing, and workflows without requiring complex prompts. As the Lead Engineer for Machine Learning, the person in this position takes charge of turning model capabilities into reliable, scalable production systems while working across the entire model lifecycle, which includes data, training, evaluation, inference, and deployment.

Key Responsibilities

Direct the end-to-end machine learning infrastructure spanning data collection, training, evaluation, inference, and deployment.
Develop and improve training and fine-tuning procedures for large models.
Create evaluation frameworks to track capability, safety, robustness, and real-world performance.
Construct high-performance inference systems to optimize latency, hardware utilization, memory, cost, and stability.
Establish data pipelines for gathering and processing real-world and synthetic training datasets.
Set up dependable production environments for deploying, tracking, and upgrading models.
Collaborate with research and application engineering teams to integrate model features into product updates.
Evaluate technical trade-offs and iterate quickly according to real-world outcomes.

Requirements

Background in developing and deploying production-grade machine learning systems rather than research prototypes.
Knowledge of modern large-model training, fine-tuning, assessment, and inference.
Foundation in software engineering and system architecture.
Practical background in running machine learning workloads at scale, particularly GPU-based infrastructures.
Technical judgment to handle ambiguous challenges independently.

Skills and Tools

Python
PyTorch
JAX
GPU-based training and inference systems

About the Company

bjakcareer operates as a compact, skilled, and hands-on team where engineers maintain broad ownership and execute tasks independently. The organization values speed, strong engineering fundamentals, and transparency, maintaining a streamlined interview process of three to four rounds conducted virtually or onsite.

How to Apply

Interested candidates should apply directly on the official company careers page.

Skills and tech stack

PythonPyTorchJAXGPU-based training and inference systemsMachine LearningFine-tuning

Questions candidates ask

Where is the Lead Engineer, Machine Learning role located?

The position is located in Indonesia.

What tools and frameworks are used in the tech stack?

The tech stack includes Python, PyTorch, JAX, and GPU-based training and inference systems.

What does the interview process involve?

The interview process consists of three to four interviews conducted via virtual meetings or onsite, evaluated by technical team members.

Is this a research-only position?

No, this is a hands-on leadership role focused on turning research and model capabilities into reliable, scalable production systems.

What is the primary mission of the company's product?

The mission is to build proactive applications that bring intelligence to conversations, errands, organizing, and workflows with minimal to no prompting.

Ready to apply?

The application is on the official bjakcareer careers page.

Apply on company site