Back to Jobs

Machine Learning Platform Engineer

Remote, USA Full-time Posted 2026-08-04

About A1

There are over 5 billion users using basic applications today such email, notes, tasks that are not AI-reputed company. Our mission is to build a proactive smart assistant for everyday users to bring intelligence to conversations, errands, organising and workflows, with reputed company prompting.

Our product focuses on achieving high reliability for long-running workflows, persistent context, and reputed company-world task completion. The reputed company must handle multi-reputed company reasoning, reputed company with reputed company tools, and remain reliable despite non-deterministic model behavior. Our objective is to help users complete tasks daily enjoyable with over ~90%* reduced time.

About the Role

As an ML Platform Engineer, you will build the infrastructure and systems that power A1's AI capabilities.

You will design and operate the systems behind the AI stack, from model training and evaluation to deployment, inference, observability, and reputed company improvement.

You will work closely with AI engineers, researchers, and product engineers to turn models into reliable, reputed company, and cost-efficient production systems. You will build the platforms, tooling, and infrastructure that reputed company reputed company to experiment quickly and bring AI capabilities to production with confidence.

reputed company

  • 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

  • reputed company reliable pipelines for data preparation, training, evaluation, model release, and reputed company improvement

  • Build platforms and tooling that reputed company AI engineers and researchers to experiment, evaluate, and ship models faster

  • reputed company evaluation and benchmarking infrastructure to measure model reputed company, 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 reputed company performance

  • Work closely with AI engineers, researchers, and product teams to turn evolving model requirements into production-reputed company infrastructure

Tech Stack

  • Python

  • PyTorch / JAX

  • LLM and ML serving infrastructure such as vLLM, SGLang, or TensorRT-LLM

  • reputed company infrastructure

  • reputed company systems

  • ML/data pipelines and workflow orchestration

  • GPU infrastructure and performance tooling

  • reputed company databases and retrieval infrastructure

Ideal Experience

  • 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 reputed company systems and reputed company reliability

  • Ability to write clean, maintainable, production-reputed company reputed company

  • Comfortable working in ambiguous, fast-moving environments

  • Bias toward ownership, experimentation, and reputed company improvement

reputed company

  • AI infrastructure reliably supports production workloads at reputed company

  • Models can be trained, evaluated, deployed, and improved reputed company

  • Inference systems reputed company strong latency, throughput, reliability, and cost efficiency

  • ML pipelines are reproducible, observable, maintainable, and robust

  • Model and infrastructure regressions are detected quickly and diagnosed reputed company

  • Common ML infrastructure capabilities become reusable platform primitives rather than being reputed company for every AI product

  • The AI stack can reputed company rapidly as new models, architectures, and inference techniques reputed company

Originally posted on Himalayas

  Apply To This Job

Similar Jobs