JobCubby JobCubby
Sign up
OnHires

Technical Lead, Machine Learning

OnHires Indonesia

Staffing and Recruiting · 11-50 employees

3 d ago
Remote machine-learning Senior (5-10 yrs) Full-time Indonesia
Create a free account to apply — email only, no card. You can also save this posting or score it against your profile with AI.

About the role

You will design and scale production machine learning systems for LLM-based applications while managing training, evaluation, and deployment pipelines. Additionally, you will optimize inference performance and mentor a high-caliber engineering team to ensure system reliability.

What they look for

Machine Learning Python PyTorch JAX LLM Fine-tuning Inference Optimization Distributed Training GPU Infrastructure Data Pipelines LoRA QLoRA SFT DPO Model Deployment Scalability Reliability

Requirements

The role requires extensive experience in building and shipping production machine learning systems using Python and PyTorch or JAX. Candidates must possess hands-on expertise in LLM fine-tuning, inference optimization, and distributed training infrastructure.

Benefits

Cash compensation Equity Remote-first environment

Full description

Technical Lead, Machine Learning

Founding AI Engineering Team | Remote | Cash + Equity

About the company

We`re partnering with ActAI, a new AI venture incubated by BJAK—one of Southeast Asia's largest and profitable technology companies. Backed by an initial $100M investment, ActAI is building the next generation of AI-native productivity applications.

The first product reimagines email using autonomous AI agents that can understand context, reason through multi-step tasks, and complete work on behalf of users while keeping them in control.

This is an opportunity to join the founding engineering team before product launch and help shape both the platform and the engineering culture.

The role

We're looking for a Technical Lead, Machine Learning to turn cutting-edge AI research into reliable production systems.

This isn't a research role. You'll own the infrastructure and execution layer that makes large language models trainable, deployable, observable and scalable in production.

You'll work across model fine-tuning, inference, evaluation, data pipelines and GPU infrastructure while partnering closely with product and backend engineers.

What you'll work on

  • Design and scale production ML systems for LLM-based applications.
  • Build training and evaluation pipelines for continuous model improvement.
  • Fine-tune foundation models using modern adaptation techniques such as LoRA, QLoRA, SFT and DPO.
  • Optimize inference performance, latency and GPU utilization.
  • Develop high-quality training datasets and evaluation frameworks.
  • Own deployment, monitoring and reliability of ML services in production.
  • Help define engineering standards while mentoring a small, high-calibre ML team.

What we're looking for

  • Experience building and shipping production ML systems used by real users.
  • Strong Python and PyTorch (or JAX) experience.
  • Hands-on experience with LLM fine-tuning and inference optimization.
  • Deep understanding of ML infrastructure, distributed training or GPU-based systems.
  • Strong engineering mindset with a focus on scalability, reliability and code quality.
  • Comfortable taking ownership in a fast-moving startup environment.

Why join ActAI?

  • Founding engineering role with significant technical ownership.
  • Opportunity to influence product architecture from the ground up.
  • Backed by $100M while operating with the speed of a startup.
  • Small, high-talent engineering team focused on solving challenging AI infrastructure problems.
  • Competitive cash compensation plus equity.
  • Remote-first environment with long-term growth opportunities.

Similar roles