Machine Learning Platform Engineer

What this role involves

The role is for an ML Platform Engineer at Bjak, focusing on building infrastructure for AI products. Responsibilities include designing systems for model training, deployment, inference, and evaluation, as well as improving reliability and cost efficiency. The tech stack includes Python, PyTorch/JAX, and various serving and cloud technologies.

Skills this posting asks for

• python
• pytorch
• jax
• vllm
• sglang
• tensorrt-llm
• cloud infrastructure
• distributed systems
• ml pipelines
• workflow orchestration
• gpu infrastructure
• vector databases
• retrieval infrastructure
• model deployment
• model inference
• model evaluation
• data pipelines
• observability
• monitoring
• tracing
• alerting

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…