Technical Lead, Machine Learning

Onhires

Europe +1 more

Remote

Who this role is best for

Aimed at senior ML engineers comfortable with productionizing LLMs in a fast-moving startup environment, with deep expertise in PyTorch/JAX and distributed training.

Best fit for

Senior ML engineers who have shipped production LLM systems at scale.
— “Experience building and shipping production ML systems used by real users.”
Technical leaders who thrive in founding teams with high ownership.
— “Founding engineering role with significant technical ownership.”
Engineers with strong Python/PyTorch skills and GPU optimization experience.
— “Strong Python and PyTorch (or JAX) experience.”

Things to consider

This is an execution-focused role, not an AI research position.
— “This isn't a research role.”
Must be comfortable working across ML ops, infra, and product integration.
— “Design and scale production ML systems for LLM-based applications.”

How to stand out

Highlight specific LLM fine-tuning techniques like LoRA or DPO.
— “Fine-tune foundation models using modern adaptation techniques such as LoRA, QLoRA, SFT and DPO.”
Demonstrate experience optimizing inference latency and GPU utilization.
— “Optimize inference performance, latency and GPU utilization.”

Pace · Fast PacedCollaboration · HighAutonomy · HighDecision Impact · TeamLevel · Senior

Derived from job-description analysis by Serendipath's career intelligence engine.

What success looks like

Design and scale production ML systems
Build training and evaluation pipelines
Develop high-quality training datasets

Typical background

Experience building production ML systemsStrong engineering mindset

Skills & requirements

Required

PythonPyTorchLLM Fine-tuningInference OptimizationGPU Infrastructure

Preferred

JAXLoraQloraSFTDPO

Stack & domain

PythonPyTorchJAXLLM Fine-tuningInference OptimizationML InfrastructureDistributed TrainingGpu-based SystemsLeadershipCommunicationTeamworkAIMachine LearningProductivity Applications

About the role

Original posting from Onhires via Ashby

TECHNICAL LEAD, MACHINE LEARNING
FOUNDING AI ENGINEERING TEAM | REMOTE | CASH + EQUITY
ABOUT THE COMPANY
We`re partnering with A1, a new AI venture incubated by BJAK—one of Southeast Asia's largest and profitable technology companies. Backed by an initial $100M investment, A1 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 A1?
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.

Source: Onhires careers (Ashby)