AI Research Engineer

The role

You will design, implement, and iterate on ML experiments for foundational model development. You will work on reinforcement learning, transformer-based architectures, and diffusion models to shape autonomous systems and intelligent robotics.

What you will do

• Design, build, and iterate on experimental ML pipelines supporting foundational model development.
• Implement and train large-scale models, including transformers and diffusion-based architectures, for generative and control tasks.
• Develop and evaluate reinforcement learning algorithms and frameworks for autonomous behaviors.
• Rapidly prototype and deploy research ideas into working code to accelerate AI experimentation cycles.
• Collaborate with cross-functional teams to integrate ML components into real-world systems.
• Stay at the forefront of the latest AI research and share insights internally and externally.

Required qualifications

• Master’s degree in Machine Learning, AI, Computer Science, or a related technical field.
• Minimum 3 years of experience in applied ML research and software engineering.
• Hands-on experience training large-scale ML models such as transformers or diffusion models.
• Strong understanding of reinforcement learning fundamentals and applications.
• Proficiency in Python and ML frameworks such as PyTorch, JAX, or TensorFlow.
• Proven ability to write clean, efficient, and scalable code that supports fast iteration.
• Excellent collaboration and communication skills.
• Willing to occasionally travel.
• Must have and maintain US work authorization.

Nice to have

• PhD in Machine Learning, AI, Computer Science, or related field.
• 4+ years of experience in applied ML research and software engineering.
• Experience designing ML experimentation frameworks and model training pipelines.
• Practical knowledge of simulation environments or robotics systems.
• Familiarity with robotic systems or simulation environments is a plus.
• Familiarity with multimodal architectures and imitation learning.
• Understanding of edge compute or distributed ML training infrastructure.

Benefits

• Relocation stipend if you are relocating to join our Seattle office and currently live more than 50 miles outside of the office location.

Before you apply

• Must have and maintain US work authorization.

About the company

Collaborative Robotics is a team of innovators and builders redefining the future of human-robot interaction. They are working to realize a world where robots are a trusted extension of your surroundings.