Senior Staff Machine Learning Engineer - AI Foundation
August 27
🏢 In-office - Bay Area
Description
• Design, train, and deploy large deep learning models that can leverage the vast amount of labeled and unlabeled data from a fleet of million vehicles. • Work with a team of talented software engineers, machine learning engineers and research scientists to push the boundary of state-of-art machine learning models which will enable the next-generation E2E solution of autonomous driving.
Requirements
• PhD in CS/CE/EE, or equivalent, in industry experience. • Deep knowledge of PyTorch. • Knowledge of model training framework (e.g. PyTorch Lightning) • In-depth knowledge of transformer architecture and ways to accelerate the training and inference of transformer models. • Experience of performing large scale distributed training of models. • A track record of profiling model and doing detective work to improve model training and inference speed.
Benefits
• A fun, supportive and engaging environment • Infrastructures and computational resources to support your work. • Opportunity to work on cutting edge technologies with the top talents in the field. • Opportunity to make significant impact on the transportation revolution by the means of advancing autonomous driving • Competitive compensation package • Snacks, lunches, dinners, and fun activities
Apply Now
