Imitation Learning Researcher (m/f/d) - #2186415

Who We Are:

We are a rapidly growing embodied AI company revolutionizing human labor. Leveraging cutting-edge robotics and advanced artificial intelligence, we develop transformative technologies that redefine how work is done across multiple industries—empowering businesses to streamline operations, boost productivity, and unlock new possibilities.

Overview:

As an Imitation Learning Researcher, you will work on developing and improving state-of-the-art imitation learning algorithms and methodologies. You’ll design experiments, create datasets, and develop models that enable systems to mimic human decision-making processes in real-world applications. This role involves close collaboration with cross-disciplinary teams to integrate findings into practical applications.

Your Responsibilities:

Design

• Design and implement advanced imitation learning algorithms.
• Develop, test, and deploy models in Python/C++/TensorRT/GGUF.
  
  

Optimize

• Analyze and preprocess datasets for training models.
• Conduct experiments to evaluate model performance and improve robustness.
• Collaborate with other researchers and engineers to integrate solutions into larger systems.
  
  

Qualifications:

Education And Experience

• Master’s or Ph.D. in Computer Science, Machine Learning, Robotics, or a related field.
  
  

Skills

• Strong proficiency in Python and PyTorch.
• Strong understanding of machine learning principles, particularly reinforcement, imitation learning, behaviour cloning, generative AI (e.g. Diffusion, Flow Matching, VAEs).
• Familiarity with teleoperation, ROS2 framework and open-source platforms such as HuggingFace, Detectron2.
• Strong ability to customize algorithm for different use-cases and multi-modality fusion.
• Strong analytical and problem-solving skills
  
  

What We Offer:

• Wellpass (gym membership)
• Free meals at the workplace
• Flexible working hours
• Option to work from home when needed
• A motivated team and an open corporate culture
• Competitive compensation and excellent career development opportunities