What are the responsibilities and job description for the Machine Learning Engineer, ML Runtime & Optimization, Mid-Level position at Jobright.ai?
Verified Job On Employer Career Site
Job Summary:
Pony.ai develops autonomous driving technology for vehicles that operates using artificial intelligence and machine learning. As a Machine Learning Engineer in ML Runtime & Optimization, you will be developing technologies to accelerate the training and inferences of the AI models in autonomous driving systems.
Responsibilities:
• Identifying key applications for current and future autonomous driving problems and performing in-depth analysis and optimization to ensure the best possible performance on current and next-generation compute architectures.
• Collaborating closely with diverse groups in Pony.ai including both hardware and software to optimize and craft core parallel algorithms as well as to influence the next-generation compute platform architecture design and software infrastructure.
• Apply model optimization and efficient deep learning techniques to models and optimized ML operator libraries.
• Work across the entire ML framework/compiler stack (e.g.Torch, CUDA and TensorRT), and system-efficient deep learning models.
Qualifications:
Required:
• BS/MS or Ph.D in computer science, electrical engineering or a related discipline.
• Strong programming skills in C/C or Python.
• Experience on model optimization, quantization or other efficient deep learning techniques.
• Good understanding of hardware performance, regarding CPU or GPU execution model, threads, registers, cache, cost/performance trade-off, etc.
• Experience with profiling, benchmarking and validating performance for complex computing architectures.
• Experience in optimizing the utilization of compute resources, identifying and resolving compute and data flow bottlenecks.
• Strong communication skills and ability to work cross-functionally between software and hardware teams.
Preferred:
• Experience with parallel programming, ideally CUDA, OpenCL or OpenACC.
• Experience in computer vision, machine learning and deep learning.
• Strong knowledge of software design, programming techniques and algorithms.
• Good knowledge of common deep learning frameworks and libraries.
• Deep knowledge on system performance, GPU optimization or ML compiler.
Company:
Pony.ai develops autonomous driving technology for vehicles that operates using artificial intelligence and machine learning. Founded in 2016, the company is headquartered in Fremont, California, USA, with a team of 1001-5000 employees. The company is currently Public Company. Pony.ai has a track record of offering H1B sponsorships.
