Principal Machine Learning Engineer

What this role involves

This is a Principal Machine Learning Engineer role at Bjak, focusing on architecting and building large-scale ML systems for an AI-native assistant product. The role involves training, inference, evaluation, and infrastructure, with emphasis on GPU optimization and production deployment. The ideal candidate has deep learning expertise and experience with distributed training and inference frameworks.

Skills this posting asks for

• deep learning
• transformer-based architectures
• pytorch
• jax
• deepspeed
• fsdp
• megatron
• zero
• ray
• gpu optimization
• quantization
• mixed precision
• vllm
• tensorrt-llm
• fastertransformer
• rlhf
• ppo
• dpo
• orpo
• multimodal models
• diffusion models
• apache arrow
• spark
• distributed training

Requirements

• Level: principal

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…