Solving the autonomous driving problem through the establishment of a state-of-art ML infrastructure for training very large foundation models and accelerating model training/inference.
Requirements
- Good knowledge of PyTorch
- Knowledge of model training framework (e.g. PyTorch Lightning)
- Knowledge of transformer architecture and ways to accelerate the training and inference of transformer models
- Experience of using pytorch ddp for distributed training of models
- Experience in training large scale vision or language models
- Previous experience in the autonomous driving industry
- Strong publications records in top academic conferences or journals, e.g. CVPR, NeurIPS, ICML, ICCV, ECCV, ICLR
Responsibilities
- 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.
Other
- Master in CS/CE/EE, or equivalent, with 5 + years of industry experience
- PhD in CS/CE/EE, or equivalent, with 3 + years of industry experience
- Being efficiently in 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
