ML Runtime Optimization Engineer - Lead
Applied Intuition
About Applied Intuition
Applied Intuition is a Tier 1 vehicle software supplier that accelerates the adoption of safe and intelligent machines worldwide. Founded in 2017, Applied Intuition delivers the definitive ADAS/AD toolchain and a world-class vehicle platform to help customers shorten time to market, build industry-leading products, and create next-generation consumer experiences. 18 of the top 20 global automakers trust Applied Intuition’s solutions to drive the production of modern vehicles. Applied Intuition serves the automotive, trucking, construction, mining, agriculture, and defense industries and is headquartered in Mountain View, CA, with offices in Ann Arbor and Detroit, MI, Washington, D.C., Munich, Stockholm, Seoul, and Tokyo. Learn more at https://appliedintuition.com.
Please note that we are an in-office company, which means the expectation is that you would come in to your Applied Intuition office 5 days a week.
About the role
We are looking for a software engineer with expertise in optimizing ML models and deploying them on production-grade runtime environments and chips. You’ll work across the entire ML framework/compiler stack (e.g. PyTorch, JAX, ONNX, TensorRT, CUDA, XLA, Triton).
At Applied Intuition, you will:
- Build the optimization pipeline for deploying ML models to real-world hardware.
- Build foundational libraries for analyzing and optimizing model performance, correctness, numerical stability, and cross-platform reproducibility.
- Closely collaborate with ML developers on model architecture details to reduce compiled latency and resource usage.
- Learn about the variety of production-grade boards our customers use and develop computational resource strategies for different customer needs.
We're looking for someone who has:
- B.Sc in Computer Science, Mathematics or a related field
- Knowledge and experience with ML accelerators, GPU, CPU, SoC architecture and micro-architecture.
- Proficiency in C++, strong software development skills with the focus on high-performance computing.
- Working experience with Python.
- Experience in developing on or using deep learning frameworks (e.g., PyTorch, JAX, ONNX, etc.)
Nice to have:
- M.Sc or PhD in a ML related area
- Built a ML compiler or optimization framework from scratch before
- Deployed ML solutions to embedded chips for real time robotics applications
The salary range for this position is $125,000 - $222,000 USD annually. This salary range is an estimate, and the actual salary may vary based on the Company's compensation practices.
Don’t meet every single requirement? If you’re excited about this role but your past experience doesn’t align perfectly with every qualification in the job description, we encourage you to apply anyway. You may be just the right candidate for this or other roles.
Applicants will be required to be fully vaccinated against COVID-19 upon commencing employment. Reasonable accommodations will be considered on a case-by-case basis for exemptions to this requirement in accordance with applicable federal and state law. Applicants should be aware that for external-facing roles that involve close contact with Company employees or other third parties on the Company's premises, accommodations that involve remaining unvaccinated against COVID-19 may not be deemed reasonable. The Company will engage in the interactive process on an individualized basis taking into account the particular position.
Applied Intuition is an equal opportunity employer and federal contractor or subcontractor. Consequently, the parties agree that, as applicable, they will abide by the requirements of 41 CFR 60-1.4(a), 41 CFR 60-300.5(a) and 41 CFR 60-741.5(a) and that these laws are incorporated herein by reference. These regulations prohibit discrimination against qualified individuals based on their status as protected veterans or individuals with disabilities, and prohibit discrimination against all individuals based on their race, color, religion, sex, sexual orientation, gender identity or national origin. These regulations require that covered prime contractors and subcontractors take affirmative action to employ and advance in employment individuals without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, protected veteran status or disability. The parties also agree that, as applicable, they will abide by the requirements of Executive Order 13496 (29 CFR Part 471, Appendix A to Subpart A), relating to the notice of employee rights under federal labor laws.