Tech Stack
Job Description, Responsibilities & Requirements
About the Position
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.
Responsibilities
- 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.
Requirements
- 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.
Locations
- European Union
- Indonesia
- Philippines
- Singapore
- Thailand
Employment Type
Full time
Location Type
Remote