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