Technical Lead, Machine Learning

About the role

Bjak is building proactive, AI-native applications to help users manage conversations, workflows, and daily tasks with minimal prompting. As a Technical Lead for Machine Learning, you will own the execution layer of the company's intelligence, transforming research and model capabilities into reliable, scalable production systems. This hands-on leadership role requires working across the entire model lifecycle, including data, training, evaluation, inference, and deployment.

Responsibilities

• Own the end-to-end ML systems powering the 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.

Required skills

• Python
• PyTorch
• JAX
• Machine Learning
• Software Engineering
• GPU-based systems
• Large model training
• Inference systems

Qualifications

• Experience building and shipping ML systems used in production
• Experience operating ML workloads at scale

About the Company

Bjak is a small, high-talent-density team focused on building proactive applications for a global user base. The company prioritizes rapid decision-making, engineering fundamentals, and building exceptional products.