AI Engineer - Diffusion Models / Flow Matching for Motion Generation

Flexion Robotics; Zürich; Vor Ort; CHF 80’000 - 100’000; Vollzeit

Zusammenfassung

A robotics software company in Zurich is seeking an expert in generative machine learning models to develop state-of-the-art multi-modal models for robot trajectory generation. The ideal candidate will have a PhD or master's degree with hands-on experience in diffusion models, excellent knowledge of Python and PyTorch, and familiarity with GPU-based simulations. This role offers competitive remuneration and the chance to take significant ownership of projects within a dynamic team.

Leistungen

Competitive remuneration

Benefits based on experience

Qualifikationen

• PhD or master's in diffusion models or trajectory generation.
• Proficient in training and fine-tuning diffusion models.
• Experience in deploying learning-based systems for robots.

Aufgaben

• Develop and deploy multi-modal models for robot trajectory generation.
• Take ownership of projects and decisions.
• Work collaboratively in a fast-paced environment.

Kenntnisse

Hands-on experience in diffusion models

Excellent knowledge of Python

Experience with PyTorch

Knowledge of machine learning architectures

Ausbildung

PhD or Master's degree in relevant field

Tools

Omniverse

Genesis

Jobbeschreibung

We are looking for an expert in generative machine learning models to strengthen our team in Zurich, particularly in diffusion models and flow matching.

The goal of this position is the development and deployment of state of the art multi-modal models for robot trajectory generation.

Requirements

Hands-on experience in the development and training of diffusion or flow matching models for robotic applications.

Particularly :

• PhD or master's degree in diffusion models / flow matching or learning-based trajectory generation with relevant project experience (e.g. master thesis).
• Excellent knowledge of Python, PyTorch and the training and fine-tuning of diffusion models and / or flow matching.
• Experience in the deployment of learning-based trajectory generation for robotic systems.
• Experience with modern GPU-based simulations such as Omniverse or Genesis.
• Good knowledge of state of the art machine learning architectures.

We are looking for a person who enjoys working in a team in a very dynamic and fast-moving environment, and who is able and willing to take ownership of projects and decisions.

We at Flexion Robotics (flexion.ai) are a young company in Zurich working on the next generation of humanoid robot software to enable robots to perform useful tasks autonomously.

We work dynamically and move fast. The team is still fairly small and every new employee at this stage will have significant ownership of their current project.

We offer a competitive remuneration in the form of a competitive base salary and additional components / benefits depending on the specific role and the experience of the applicant.