ML Platform Engineer for AI Infra & Pipelines

Bjak; Spain (TX); On-site; USD 140,000 - 200,000; Full time

Job summary

A1 is seeking an ML Platform Engineer to build the infrastructure powering AI capabilities. You will design and operate systems behind the AI stack, from training and evaluation to deployment, inference, observability, and continuous improvement.

You will collaborate with AI engineers, researchers, and product teams to turn evolving model requirements into reliable, scalable production platforms, tooling, and pipelines that enable rapid experimentation and cost-efficient operation.

Qualifications

• 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.

Responsibilities

• Build and operate ML infrastructure and platforms powering AI products.
• Design systems for 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, model release, and continuous improvement.
• Build platforms and tooling enabling AI engineers and researchers to ship models faster.
• Develop evaluation and benchmarking infrastructure to measure model quality and regressions.
• Create production observability, monitoring, tracing, and alerting for ML workloads.
• Identify bottlenecks across the ML stack and drive performance improvements.

Skills

Software Eng

ML infra

Model deployment

Distributed systems

Clean code

Ambiguity tolerance

Ownership mindset

PyTorch

TensorRT-LLM

vLLM

Tools

PyTorch

TensorRT-LLM

vLLM

SGLang

Job description

A1 is seeking an ML Platform Engineer to build the infrastructure powering AI capabilities. You will design and operate systems behind the AI stack, from training and evaluation to deployment, inference, observability, and continuous improvement.

You will collaborate with AI engineers, researchers, and product teams to turn evolving model requirements into reliable, scalable production platforms, tooling, and pipelines that enable rapid experimentation and cost-efficient operation.