Software Engineer - ML Framework

About the role

Applied Intuition is pushing the frontier of autonomy and vehicle intelligence through next-generation AI Research. In this role, you will enable this research effort by building ML tools, infrastructure, managing large datasets, and addressing challenges across the entire end-to-end (E2E) autonomy stack. Join our research and production teams in creating the next generation of self-driving vehicles. Here at Applied, 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:

• Build tools and infrastructure to enable large-scale E2E autonomy research led by Dr. Wei Zhan

• Work directly with both AI research and engineering teams to productionize next generation self-driving software

• Help scale end-to-end training by identifying bottlenecks, especially as we continue to scale our GPU compute, onroad data volume, curation efforts, and eval systems

• Technologies: Python, Pytorch, CUDA, Bazel, Kubernetes, Spark, Flyte

We're looking for someone who has:

• Exceptional software generalist experience with ambition to contribute up and down the ML stack, onboard the vehicle, and in the cloud

• Strong fundamental skills in software & module design: can tame chaos with libraries & abstractions

• Rapid analytical and problem-solving skills

• Bachelors, Masters or PhD degree in any field, or equivalent combination of education and work experience.

• 3+ years of experience in any field, or equivalent combination of education and experience

Nice to have:

• Strong multi-language build system knowledge

• Distributed systems/compute (e.g. k8s, cloud-managed services at scale)

• ML infra fluency - has used and/or contributed to ML and large data systems

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.