Note: The job is a remote job and is open to candidates in USA. BJAK is building proactive, AI-native applications that help users manage conversations, errands, organization, and workflows with minimal prompting. The Technical Lead, Machine Learning owns the execution layer of the company’s intelligence, leading end-to-end machine learning systems across data, training, evaluation, inference, and deployment while translating research and model capabilities into reliable production systems.
Responsibilities
- Own the end-to-end ML systems powering our company, from data and training to evaluation, inference, and deployment
- Build and evolve training and fine-tuning pipelines for large models
- Design evaluation systems that measure capability, robustness, safety, and real-world product performance
- Architect high-performance inference systems, optimizing latency, GPU utilization, memory, cost, and reliability
- Build data pipelines and systems for high-quality real-world and synthetic training data
- Establish reliable production infrastructure for deploying, monitoring, and continuously improving models
- Partner closely with research and application engineering to turn model capabilities into product improvements
- Make pragmatic technical trade-offs and rapidly iterate based on real-world performance
Skills
- • Experience building and shipping ML systems used in production, not just research prototypes
- • Strong understanding of modern large-model training, fine-tuning, evaluation, and inference
- • Strong software engineering and systems fundamentals
- • Experience operating ML workloads at meaningful scale, particularly GPU-based systems
- • Strong technical judgment and the ability to navigate ambiguous problems independently
- • A bias toward experimentation, measurement, and shipping
- • High standards for correctness, reliability, and production quality
- • Python
- • PyTorch / JAX
- • GPU-based training and inference system
- • You have built or shipped real ML systems used by people, not just demos
- • You are comfortable working with large models and understanding their failure modes
- • You write strong, production-grade code and care about system correctness
Benefits
- Remote work arrangement
Company Overview