AI Engineer - Diffusion Models

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.

Benefits

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.