Technical Lead, Machine Learning

What this role involves

Bjak is hiring a Technical Lead for Machine Learning to own the execution layer of their AI assistant product. The role involves building production-grade ML systems, fine-tuning models, and collaborating with application engineering. The ideal candidate has shipped real ML systems and is comfortable with large models.

Skills this posting asks for

• python
• pytorch
• jax
• gpu
• lora
• qlora
• sft
• dpo
• distillation
• ml pipelines
• data pipelines
• training workflows
• evaluation systems
• inference architecture
• deployment
• gpu optimization
• memory efficiency
• latency reduction
• scaling policies

Requirements

• Level: lead

From the employer’s posting

ABOUT A1 There are over 5 billion users using basic applications today such email, notes, tasks that are not AI-native. Our mission is to build a proactive smart assistant for everyday users to bring intelligence to conversations, errands, organising and workflows, with minimal prompting. Our pro…