1943 IT & Software Developer jobs in the US
Senior Machine Learning Engineer - Palo Alto
$162,800 - 205,300
Rivian
Hansen Way 607, Palo Alto
$162,800 - 205,300
Requirements
Must:
- Ph.D. or M.S. in Computer Engineering, Electrical Engineering, Computer Science, or a related field with a specialization in ML compilers, embedded systems, or hardware-aware AI.
- Hands-on experience with deploying quantized models, ML design frameworks, and code generation for embedded or heterogeneous computing environments.
- Strong knowledge of computer vision models, such as object detection and segmentation, and their optimization for edge inference.
- Proficiency in deep learning frameworks like PyTorch and TensorFlow and familiarity with their low-level intermediate representations or export formats such as ONNX.
- Solid programming experience in C++ and Python.
- Familiarity with CUDA/OpenCL or similar programming models for accelerators.
- Previous involvement in hardware-software co-design, particularly for autonomous systems or robotics is preferred.
- Understanding of numerical precision trade-offs, quantization-aware training (QAT), and dynamic/static quantization techniques is a plus.
- Knowledge of embedded real-time constraints and associated profiling/debugging tools is advantageous.
- Experience in rearchitecting models to optimally fit hardware capabilities.
- A publication record in reputable ML/Systems conferences is desirable.
Technologies
AI
CUDA
PyTorch
Robotics
Responsibilities
- Conduct research on state-of-the-art perception models in collaboration with ADAS software teams.
- Drive the development of optimization strategies for deploying quantized perception models such as CNNs, Transformers, and LLMs onto various embedded and heterogeneous hardware platforms.
- Create and apply hardware-aware optimization techniques, which include quantization strategies, model compression, memory-efficient representations, and operator fusion tailored for custom accelerators.
- Work alongside hardware teams to co-optimize model architectures and compute pipelines while adhering to real-time requirements like latency, throughput, and power consumption.
- Measure and evaluate system performance across different platforms, iterating to achieve optimal deployment outcomes.
- Collaborate with perception, systems, and autonomy teams to ensure that model optimization aligns with the hardware roadmap and the practical demands of autonomy.
Description
At Rivian, we are dedicated to fostering adventure in a sustainable manner. We pride ourselves on the innovative spirit of our team, which comprises individuals from various backgrounds who share a passion for the outdoors and a commitment to protecting it for future generations. Our workplace is filled with opportunities for growth and collaboration. We provide comprehensive medical, dental, and vision insurance packages for full-time and part-time employees and their families, with coverage starting on the first day of employment for full-time employees. Located in Palo Alto, CA, we offer competitive salaries and are committed to equal opportunity employment, fostering an inclusive environment where all qualified applicants are considered regardless of their background or identity.
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 1943 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?
Sequoia Software, D3 Security Management Systems, Snaplii, Cobmex® Apparel Inc., KUBRA, Nurse Next Door, Allied Technical Solutions among others, are currently hiring for ML, AI roles in the United States.
The company with most openings is Jobot as they are hiring for 124 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 Jobot as they are hiring for 124 different Machine Learning Engineer jobs in the United States. They are probably quite committed to find good Machine Learning Engineers.