Technical Lead, Machine Learning
About the Company
A1 is 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, starting with an autonomous AI agent that reimagines email.
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
- Ability to take ownership in a fast-moving startup environment
Benefits
- 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
- Competitive cash compensation plus equity
- Remote-first environment
