Machine Learning Engineer, Reinforcement Learning

Company Overview:

Skild AI is a startup focused on creating large-scale foundation models for robotics with the goal of developing general-purpose robotic intelligence. We view data-driven machine learning methods as the key to unlocking general-purpose capabilities for the widespread deployment of robots to perform economically useful tasks within society. Our team consists of individuals with varying levels of experience and backgrounds, from new graduates to domain experts. Relevant industry experience is important, but ultimately less so than your demonstrated abilities and attitude. We are looking for passionate individuals who are eager to explore uncharted waters and contribute to our innovative projects. Both generalists and specialists within specific sub-fields of robotics (planning, SLAM, manipulation, computer vision, etc.) are encouraged to apply.

Position Overview:

We are looking for a Machine Learning Engineer to be responsible for designing and implementing cutting-edge reinforcement learning algorithms, conducting experiments, and optimizing these models to perform efficiently in real-world robotic environments. This will require close collaboration with our robotics, research and engineering team. Your work will directly impact the development of intelligent, adaptable robots capable of learning and performing complex tasks autonomously. 

Responsibilities:

• Develop and implement state-of-the-art reinforcement learning algorithms for robotic applications.
• Design and conduct experiments to train RL models and conduct real-world tests.
• Collaborate closely with researchers to explore novel methods of scaling up reinforcement learning model training. 
• Communicate effectively with inference, application, and deployment engineers to integrate RL models into robotic systems and iterate on methods to enable robust deployment. 
• Analyze and interpret experimental results, iterating on model design to achieve desired performance.
• Stay up-to-date with the latest research and advancements in reinforcement learning. 

Preferred Qualifications:

• BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field, or equivalent practical experience. 
• Proficiency in Python, C++, or similar and at least one deep learning library such as PyTorch, Tensorflow, JAX, etc. 
• Deep understanding and practical experience with various reinforcement learning algorithms anf techniques (model-free, model-based, multi-task, hierarchical, multi-agent, etc.).
• Strong background in algorithms, data structures, and software engineering principles.
• Experience with physics simulation engines and tools for training RL. 
• Deep understanding of state-of-the-art machine learning techniques and models. 
• Extensive industry experience with reinforcement learning and robotic systems.