Senior/Staff ML Engineer, ML Acceleration and Performance

Rivian
US - California - Palo Alto
View Company Profile / << Go Back

  • Job Type: Full time
  • 7 days ago

Job Description

Develop and optimize ultra-low latency machine learning algorithms for safety-critical self-driving features. Bridge the gap between theoretical ML research and real-time embedded deployment using hardware-aware optimization strategies.

Requirements: Requires an MS with 3+ years of experience or a PhD in Computer Science or Electrical Engineering. Must have deep expertise in modern model architectures, model compression techniques, and GPU optimization.

Key Skills: Model Optimization, Hardware Acceleration, PyTorch, TensorFlow, CUDA, C++, TensorRT, Quantization, Model Distillation, GPU Architecture, Python, ONNX, Performance Profiling, Deep Learning, Transformers, LLMs

Benefits: Medical Insurance, Prescription Drug Insurance, Dental Insurance, Vision Insurance




Fast Track Upload