ML Platform Engineer: Build Scalable AI Infrastructure

ActAI; Sweden; On-site; SEK 700,000 - 1,100,000; Full time

Job summary

ActAI is seeking an ML Platform Engineer in Sweden to build the infrastructure behind our AI capabilities. You will design, deploy, and operate systems for model training, evaluation, deployment, and inference, working with AI researchers and product engineers to productionize models.

You will own platforms and tooling to enable rapid experimentation, ensure reliability and observability, and drive cost-efficient, scalable production workflows for AI products.

Qualifications

• Strong software engineering fundamentals.
• Experience building ML infrastructure or production ML systems.
• Experience with model deployment, inference, evaluation, or data pipelines.
• Knowledge of distributed systems and system reliability.

Responsibilities

• Build and operate ML infrastructure and platforms powering AI products.
• Design systems for training, evaluation, deployment, inference, and experimentation.
• Build scalable serving and inference infra for high-throughput workloads.
• Improve reliability, latency, and cost efficiency of AI systems.
• Develop reliable data pipelines and model release processes.
• Provide observability and tooling for AI workloads.

Skills

Python

Distributed systems

Production-grade code

Ambiguity handling

Tools

PyTorch

JAX

vLLM

TensorRT-LLM

Cloud infrastructure

GPU tooling

Job description

ActAI is seeking an ML Platform Engineer in Sweden to build the infrastructure behind our AI capabilities. You will design, deploy, and operate systems for model training, evaluation, deployment, and inference, working with AI researchers and product engineers to productionize models.

You will own platforms and tooling to enable rapid experimentation, ensure reliability and observability, and drive cost-efficient, scalable production workflows for AI products.