GPU Lead, Machine Learning Platform

Key Skills

Python Programming, Scala Programming, C++, Java Programming, Machine learning techniques, Data science techniques, SQL, Apache Hadoop, MapReduce, TensorFlow, PyTorch

Job Description

The Machine Learning Infrastructure team builds the industrial scale machine learning platform used by ML practitioners developi ng software for our self-driving car. We provide essential tools and frameworks to support the entire lifecycle of machine learning, from data processing, large scale training and evaluation frameworks to efficient neural net inference runtimes for onboard execution and simulation.

The Motional Santa Monica office is located in the heart of 3rd Street Promenade at 1453 3rd Street. With all the modern amenities f or productive work and play, it is steps away from the popular pedestrian outdoor shopping area and Santa Monica beach.

What You'll Be Doing

• Define, design and implement the compute platform to: Manage and monitor ML training, Visualize and compa re ML training, Evaluate model on different hardware accelerators and Track cost etc.
• Own training cluster and provide technical support to internal customers.
• Inspire and motivate the team to work together as a cohesive and productive unit
• Work closely with ML researchers to identify the bottlenecks and optimize the codebase by: Better paralle lism and data caching, Fully utilizing the hardware: CPU, GPU, memory and disk/network IO, Distributed/multi-machine training.
• Act as a sounding board and technical architect for your teams, help define technical vision and strategy
• Provide coding/performance guidelines and best practices to ML researchers/engineers.

What We're Looking For

• Degree in Computer Engineering, Computer Science, Applied Mathematics, or a related field
• 5+ years of industry experience, including 3+ years as Tech Lead or Manager
• Strong programming skills in Python.
• Familiar with DL framework like PyTorch, Tensorflow, Keras etc.
• Experience working with cloud infrastructure like AWS, GCP or Azure.
• Experience working with large scale dataset.

Bonus Points

• Strong skills in C++.
• Experience in numba, cython, CUDA etc.
• Experience in distributed computing.
• Strong ML/DL background.

Apply Online at Motional