Research Engineer, ML Pipelines & Distributed Training

Clera; Zürich; On-site; CHF 120,000 - 180,000; Full time

Benefits offered by this job

Equity

Health, dental, and vision coverage

Professional development support

Visa sponsorship

Relocation assistance

Job summary

Clera in Zürich, Switzerland, seeks an experienced AI software engineer to turn experimental code into reliable, production-ready systems. You will design and build post-training pipelines for generating, processing, and evaluating research data to scale experiments.

You will collaborate with researchers to translate ideas into robust, maintainable software, optimize GPU utilization, profiling, and data throughput, and contribute to testing and experiment tracking to preserve result integrity.

Qualifications

• 2 to 10+ years of experience building ML training, inference, or data pipelines in Python.
• Strong Python skills and hands-on experience with PyTorch, JAX, or similar frameworks.
• Experience in building and debugging software for model training, inference, or data processing at scale.
• Understanding of ML experiments and how data, precision, and implementation affect results.
• Sound software engineering practices: testing, profiling, version control, and documentation.
• Experience with distributed tools (PyTorch distributed, DeepSpeed, Ray) is helpful.

Responsibilities

• Convert research prototypes into tested, reusable data generation, training, and evaluation pipelines.
• Build distributed experiment support for data loading, checkpointing, and collecting agent interactions.
• Profile workloads and improve GPU utilization, memory efficiency, and data throughput.
• Develop tests, experiment tracking, and debugging tools while preserving experimental results.
• Collaborate with researchers to turn ideas into robust, maintainable software.

Skills

Python programming

Machine learning pipelines

Tools

PyTorch

JAX

Transformers libraries

DeepSpeed

Ray

vLLM

Experiment tracking tools

Job description

Clera in Zürich, Switzerland, seeks an experienced AI software engineer to turn experimental code into reliable, production-ready systems. You will design and build post-training pipelines for generating, processing, and evaluating research data to scale experiments.

You will collaborate with researchers to translate ideas into robust, maintainable software, optimize GPU utilization, profiling, and data throughput, and contribute to testing and experiment tracking to preserve result integrity.