Principal AI Research Engineer

About the role

As a Principal AI and ML fundamentalist who is an expert at developing cutting-edge AI solutions to complex problems, you will develop code that eventually integrates into production and be responsible for:

Responsibilities

• Design, implement, and evaluate novel machine learning and deep learning algorithms.
• Iterate on AI model development; starting from the data needed for training, architecture, input/output representations, evaluation, and deployment.
• Collaborate with other researcher engineers to prototype and validate complex solutions from academic literature.
• Conduct experiments to benchmark new techniques and evaluate model behavior.
• Develop tools and frameworks to support scalable and reproducible research and development.
• Communicate research findings to leadership in a concise, and convincing manner.
• Stay current with the latest developments in AI/ML and identify relevant innovations.
• Assist in transitioning research prototypes into production-ready systems.

Requirements

• M.S or PhD degree in Computer Science, or Computer Engineering with strong emphasis on AI and ML.
• Enjoys solving less-defined and complex problems.
• Familiarity with good SW practices and development, including code quality, version control, etc.
• Very strong with ML frameworks such as Pytorch, or Tensorflow.
• Good understanding of Transformer architectures, attention mechanisms, multi-modal foundation models, diffusion policies, fine-tuning, model distillation, mixture of experts, and other similar topics.
• Has strong ability to debug and determine requirements for AI models in terms of data, training, and evaluation.
• Is up to date with research literature and recent ML techniques and methodologies.

Bonus Qualifications

• Experience with Reinforcement Learning, self-supervised training, and imitation learning is a plus.
• Experience developing VLA (Vision Language Action) models for robotic manipulation and mobility is a plus.
• Experience developing production AI models is a plus.
• Publications at top conferences are a plus.

Pay

For this position we are targeting a base pay between $270,000.00 - $340,000.00. Actual compensation offered will be determined by factors such as job-related knowledge, skills, and experience.