Machine Learning Computing Intern
Sunnyvale, CA
Mercedes-Benz R&D North America
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.
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
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
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.