Senior ML Infra Engineer – GPU Clusters (Zurich)

Roboticsconnected; Zürich; Presencial; CHF 140.000 - 190.000; Jornada completa

Ventajas ofrecidas por este puesto de trabajo

Competitive compensation

Robotics team exposure

Collaborative culture

Descripción de la vacante

Flexion is seeking a seasoned ML Engineer to own the GPU compute platforms and drive the design, bring-up, operation, and optimization of high-performance clusters on-site in Zürich. You will collaborate with AI engineers to accelerate training speed, improve hardware utilization, and shape future compute strategies across multi-cloud environments.

You will contribute to core ML libraries, build scalable tooling, and ensure reliable, cost-efficient systems while aligning with Flexion’s ambitious

Formación

• Experience with training or inference of large models (billions of parameters) on distributed multi-node GPU hardware.
• Proficiency in Python and working knowledge of PyTorch.
• Deep understanding of distributed training concepts (DDP, FSDP, NCCL).
• Experience with at least one cloud platform (AWS, GCP, Azure) or large-scale on-premises GPU infrastructure.
• Experience with job scheduling and orchestration tools: Slurm and/or Kubernetes/KubeRay.

Responsabilidades

• Architect, run and continuously improve existing and future cloud-based GPU clusters. Select the best frameworks and tooling to run our clusters efficiently. Work on cluster provisioning, job schedulers and monitoring systems.

Conocimientos

Python

PyTorch

Distributed training

Slurm

Kubernetes

Cloud platforms

Educación

Degree in Computer Science / Electrical Engineering / Software Engineering

Herramientas

AWS

GCP

Azure

Descripción del empleo

Flexion is seeking a seasoned ML Engineer to own the GPU compute platforms and drive the design, bring-up, operation, and optimization of high-performance clusters on-site in Zürich. You will collaborate with AI engineers to accelerate training speed, improve hardware utilization, and shape future compute strategies across multi-cloud environments.

You will contribute to core ML libraries, build scalable tooling, and ensure reliable, cost-efficient systems while aligning with Flexion’s ambitious