September 7
🏢 In-office - Bay Area
• 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.
• 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. • Previous experience in the autonomous driving industry. • Experience with cuda or Triton language for writing custom ops. • Knowledge of Nvidia TensorRT. • Experience with edge computing systems. • A track record of efficiently solving complex problems collaboratively on larger teams
• 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