Machine Learning Computing Intern

Sunnyvale, CA

Mercedes-Benz R&D North America

View all jobs at Mercedes-Benz R&D North America

Apply now Apply later

At Mercedes-Benz Research & Development North America (MBRDNA), we are committed to delivering world-class automotive technologies that push the boundaries of what is possible. Our teams of highly skilled engineers and designers use cutting-edge software and technology, to enhance the driving experience and reduce environmental impact. 
This internship position offers an opportunity to work on cutting-edge machine learning deployment solutions. You will be developing and optimizing machine learning models for edge devices using modern compilation frameworks like Apache TVM and Modular AI.
During this internship you will learn hardware-specific optimization techniques, experience state-of-the-art ML compilation framework, build proficiency in performance optimization and hardware abstraction techniques. 

Job Responsibilities:

  • Implement and optimize deep learning models using Apache TVM and Modular's MAX platform.
  • Create automated pipelines for model compilation targeting various hardware platforms.
  • Develop hardware-agnostic solutions for seamless deployment across devices.
  • Write efficient code in C++ and Python.
  • Optimize model inference using advanced compiler techniques.

Minimum Qualifications:

  • Master’s or PhD or Minimum level of education required and required field of study Masters in computer science. 
  • Strong programming skills in C++ and Python.
  • Familiarity with ML frameworks (PyTorch, TensorFlow, ONNX).
  • Understanding of ML compilation frameworks (Apache TVM, MLIR, Modular).
  • Experience with ML model deployment and optimization.
  • Understanding of model quantization and compression techniques.
  • Knowledge of hardware abstraction principles and edge computing.

Preferred Qualifications:

  • Experience with CUDA Programming and GPU architecture.
  • Hands-on experience deploying ML models on embedded devices.
  • Experience with real-time inference and performance optimization.
  • Knowledge of deployment workflows and CI/CD pipelines for edge devices.
Additional Information:The current hourly rate for this position is as follows and may be modified in the future: $28 (Undergraduate Students)/$32 (Graduate Students)
Benefits/Perks:•PTO•Sick Time
Why should you apply?Here at MBRDNA, you create digital ecosystems around cars, you design a language between humans and machines, you make a car even more intelligent - you make the new reality for cars. MBRDNA was honored as one of the "Best Places to Work" by BuiltIn in January 2024, a testament to our commitment to creating an exceptional work environment. At each of our offices, we foster a culture of diversity, collaboration, and continuous learning, ensuring every team member can thrive and innovate.
Benefits for Full-Time* Employees Include:• Medical, dental, and vision insurance for employees and their families• 401(k) with employer match• Up to 18 company-paid holidays• Paid time off (unlimited for salaried employees), sick time, and parental leave• Tuition assistance program• Wellness/Fitness reimbursement programs• Vehicle lease subsidy or company car (for eligible employees only)* Internships & Contractors excluded from Full-Time Employee benefits
Apply now Apply later
Job stats:  10  2  0

Tags: Architecture CI/CD Computer Science CUDA Deep Learning GPU Machine Learning ML models Model deployment Model inference ONNX PhD Pipelines Python PyTorch R&D Research TensorFlow

Perks/benefits: 401(k) matching Career development Fitness / gym Health care Medical leave Parental leave Unlimited paid time off Wellness

Region: North America
Country: United States

More jobs like this