AI Research Engineer

Overview

Job Description

Responsibilities

• - Building robust ML pipelines for training, evaluation, and deployment of control algorithms
• - Working with researchers to implement cutting-edge methods in Deep Learning and Reinforcement Learning
• - Collaborating with software engineers to integrate models into real-time systems
• - Designing tools and infrastructure to support large-scale experimentation and iteration
• - Performing rigorous evaluations on simulation and real-world data, and tuning models for safety, performance, and robustness
• - Supporting reproducibility and code quality across the team

Required Skills

• - 1+ years of prior experience in applied machine learning or software engineering
• - Bachelor''s or Master''s degree in Computer Science, Engineering, or related technical field
• - Proficiency in Python and experience with ML frameworks such as PyTorch, TensorFlow, or JAX
• - Strong foundations in supervised learning, reinforcement learning, or control systems
• - Familiarity with common tools for experimentation (e.g., Ray, MLFlow, WandB, or similar)
• - Ability to write high-quality, testable, and maintainable code
• - Share our company values: Collaboration, Transparency, Operational Excellence, Ownership, and Empathy

Benefits

• - Fast-paced and team-oriented environment
• - Competitive compensation & equity
• - Outsized responsibilities & professional development
• - Training is foundational; functional, customer immersion, and development training
• - Medical, dental, and vision insurance (exact benefits vary by region)
• - Unlimited paid time off, with a minimum of 20 days off per year requirement
• - Paid parental leave (exact benefits vary by region)
• - Home office setup allowance, coworking space stipend, and company MacBook