Staff Machine Learning Engineer - Foundation Model

We are looking for a full-time Machine Learning Engineer to establish a state-of-art ML infrastructure for training large foundation models and accelerating model training/inference. You will work with a team to push the boundary of state-of-art machine learning models for autonomous driving.

Requirements

• Master in CS/CE/EE, or equivalent, with 5 + years of industry experience.
• 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.

Benefits

• 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