Machine Learning Platform Engineer

Stack

Responsibilities

• ▹Build and operate the ML infrastructure and platforms powering A1's AI products
• ▹Design systems for model training, evaluation, deployment, inference, and experimentation
• ▹Build and optimize model-serving and inference infrastructure for high-throughput, low-latency workloads
• ▹Improve reliability, scalability, latency, and cost efficiency of AI systems
• ▹Develop reliable pipelines for data preparation, training, evaluation, and model release
• ▹Build platforms and tooling that let AI engineers and researchers experiment and ship faster
• ▹Develop evaluation and benchmarking infrastructure to measure model quality and regressions
• ▹Build production observability, monitoring, tracing, and alerting for AI/ML workloads
• ▹Identify bottlenecks and continuously improve performance across the ML stack
• ▹Work closely with AI engineers, researchers, and product teams

Requirements

• ▹Strong software engineering fundamentals, experience building production systems
• ▹Experience building ML infrastructure, platforms, or production machine learning systems
• ▹Experience with model deployment, inference, evaluation, or data pipelines
• ▹Strong understanding of distributed systems and system reliability
• ▹Ability to write clean, maintainable, production-quality code
• ▹Comfortable working in ambiguous, fast-moving environments

Nice to have

• ▹Python
• ▹PyTorch / JAX
• ▹LLM and ML serving infrastructure (vLLM, SGLang, TensorRT-LLM)
• ▹Cloud infrastructure
• ▹Distributed systems
• ▹ML/data pipelines and workflow orchestration
• ▹GPU infrastructure and performance tooling
• ▹Vector databases and retrieval infrastructure

Soft skills

About the company

A1 (Bjak) is building a proactive AI assistant for everyday users, aiming to bring intelligence to conversations, errands, organizing, and workflows with minimal prompting.