[Remote] Machine Learning Platform Engineer

Note: The job is a remote job and is open to candidates in USA. BJAK is focused on building a proactive smart assistant for everyday users, aiming to enhance productivity through AI capabilities. As an ML Platform Engineer, you will design and operate the systems behind the AI stack, working closely with engineers and researchers to create reliable and scalable production systems.

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 optimise model serving and inference infrastructure for high-throughput and low-latency workloads
• Improve reliability, scalability, latency, and cost efficiency of AI systems
• Develop reliable pipelines for data preparation, training, evaluation, model release, and continuous improvement
• Build platforms and tooling that enable AI engineers and researchers to experiment, evaluate, and ship models faster
• Develop evaluation and benchmarking infrastructure to measure model quality, performance, and regressions
• Build production observability, monitoring, tracing, and alerting for AI/ML workloads
• Improve AI systems across reliability, scalability, latency, throughput, and cost
• Identify bottlenecks across the ML stack and continuously improve system performance
• Work closely with AI engineers, researchers, and product teams to turn evolving model requirements into production-ready infrastructure

Skills

• Strong software engineering fundamentals and 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
• Bias toward ownership, experimentation, and continuous improvement

Company Overview