4574 IT & Software Developer jobs in the US
Machine Learning Compiler Engineer
$228,000 - 285,000
Rivian
Hansen Way 607, Palo Alto
$228,000 - 285,000
Requirements
Must:
- Ph.D. or M.S. in Computer Engineering or a related field.
- Strong programming skills in C/C++ and Python.
- Experience with various SOC platforms utilized for machine learning applications.
- In-depth understanding of deep learning software models.
- Previous experience in developing compiler pipelines is preferred.
- Proficiency in deep learning frameworks and their low-level IRs or export formats.
- Background in high-pressure design environments is preferred.
- Familiarity with hardware-software co-design, particularly in autonomous or robotics applications (preferred).
- Knowledge of numerical precision trade-offs, quantization-aware training (QAT), and quantization flows (preferred).
- Understanding of embedded real-time constraints and hardware profiling/debugging tools (preferred).
- Experience in rearchitecting models to optimize compatibility with hardware capabilities (preferred).
Responsibilities
- Spearhead the development of an ML Compiler for adapting Autonomy ML models to the Rivian Autonomy Processor (RAP1).
- Design and execute hardware-aware optimizations, such as quantization approaches, model compression, memory-efficient formats, and operator fusion tailored to RAP1.
- Collaborate with hardware teams to jointly optimize model architecture and compute pipelines within real-time constraints (latency, throughput, power).
- Perform benchmarking and performance analysis across different platforms, iterating to attain optimal deployment efficiency.
- Work in partnership with autonomy teams to synchronize model optimization efforts with hardware timelines and practical autonomy needs.
Description
At Rivian, we are dedicated to redefining adventure without emissions through our innovative Electric Adventure Vehicles. Our team embraces diversity and shares a passion for the outdoors, working diligently to safeguard it for generations to come. As part of the ML Compiler team, you will play a crucial role in developing software tools that facilitate deep learning network inference on our hardware platforms. We offer a comprehensive benefits package, including paid vacation and health insurance, as well as opportunities for career growth and involvement in impactful projects. Join us as we continue to explore new frontiers in technology.
Something wrong or incorrect with this job? Tell us in the chat 💬 on the right ➡️
You can find Machine Learning Engineer salaries in the United States here.
How many Machine Learning Engineer jobs are in the United States?
Currently, there are 4574 ML, AI openings. Check also: TensorFlow jobs, Python jobs, Computer-Vision jobs - all with salary brackets.
Is the US a good place for Machine Learning Engineers?
The US is one of the best countries to work as a Machine Learning Engineer. It has a vibrant startup community, growing tech hubs and, most important: lots of interesting jobs for people who work in tech.
Which companies are hiring for Machine Learning Engineer jobs in the United States?
GE Aerospace, D3 Security Management Systems, Execusource, Inc, Peraton, Triumph Tech, LLC, Dynamic Animation Systems, Inc., Rocket EMS among others, are currently hiring for ML, AI roles in the United States.
The company with most openings is Judge Group, Inc. as they are hiring for 362 different Machine Learning Engineer jobs in the United States. They are probably quite committed to find good Machine Learning Engineers.
The company with most openings is Judge Group, Inc. as they are hiring for 362 different Machine Learning Engineer jobs in the United States. They are probably quite committed to find good Machine Learning Engineers.