Research Engineer - AI Infrastructure

About the role

We are looking for both infrastructure engineers with expertise in machine learning pipelines and AI research engineers that want to work beyond modeling to join our AI Infrastructure group. This role will work across the entire AI lifecycle (dataset generation, training frameworks, compute, evaluation, and deployment) and work directly with modeling teams. This team is a good fit if you are excited to work on broad, ambiguous problems and develop across the entire ML stack. At Applied Intuition, we encourage all engineers to take ownership over technical and product decisions, closely interact with external and internal users to collect feedback, and contribute to a thoughtful, dynamic team culture.

At Applied Intuition, you will:

• Design and build training and evaluation infrastructure to support our current AI research directions, orchestrating massive GPU clusters to process PBs of multimodal sensor data
• Create robust benchmarking and continuous evaluation systems to track model progress across diverse, real-world driving distributions
• Develop scalable systems for large-scale data sampling, dataset creation, and advanced curation techniques leveraging state-of-the-art AI models
• Work across cloud environments to support high-throughput distributed training
• Collaborate closely with the AI research team and autonomy teams
• Technologies: Pytorch, CUDA, Ray, Flyte, K8s

We're looking for someone who has:

• Experience with building software components to address production, full-stack machine learning challenges.
• Opinions about building a company-wide platform for ML training, evaluation, and deployment
• Knowledge of the open source landscape with judgment on when to choose open source versus build in-house
• Excellent analytical and problem-solving skills

Compensation at Applied Intuition for eligible roles includes base salary, equity, and benefits. Base salary is a single component of the total compensation package, which may also include equity in the form of options and/or restricted stock units, comprehensive health, dental, vision, life and disability insurance coverage, 401k retirement benefits with employer match, learning and wellness stipends, and paid time off. Note that benefits are subject to change and may vary based on jurisdiction of employment.

Applied Intuition pay ranges reflect the minimum and maximum intended target base salary for new hire salaries for the position. The actual base salary offered to a successful candidate will additionally be influenced by a variety of factors including experience, credentials & certifications, educational attainment, skill level requirements, interview performance, and the level and scope of the position.

Please reference the job posting’s subtitle for where this position will be located. For pay transparency purposes, the base salary range for this full-time position in the location listed is: $153,000 - $222,000 USD annually.