Key skills
PythonPyTorchC++CAIMachine LearningGenerative AI
About this role
Role Overview
- Design and implement advanced imitation learning algorithms.
- Develop, test, and deploy models in Python/C++/TensorRT/GGUF.
- 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.
Requirements
- Master’s or Ph.D. in Computer Science, Machine Learning, Robotics, or a related field.
- 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
Tech Stack
- Python
- PyTorch
Benefits
- Wellpass (gym membership)
- 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