Staff Machine Learning Engineer

What this role involves

This is a Staff Machine Learning Engineer role at Bjak, focusing on building AI-native applications. The role involves owning end-to-end ML system execution, including data pipelines, training, evaluation, and deployment. The ideal candidate has experience shipping real ML systems and is comfortable with large models and production constraints.

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
• backend
• mobile
• desktop

Requirements

• Level: staff

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…