ML Infrastructure Tech Lead, Nuro

• 6+ years of work or research experience in ML Infra, distributed training, model inference or GPU programming
• Knowledge in using at least one deep learning framework, e.g. Tensorflow, Pytorch, JAX. Be able to understand deep learning algorithms, e.g. computer vision, NLP, behavior planning
• Proficient in C++/C and Python. Strong coding ability
• Strong communication and teamwork skills. Be passionate about exploring and promoting cutting edge technology

Desirable

• Our team provides solutions to empower machine learning development in Nuro and optimize on-cloud training and onboard inference
• Our solutions include a distributed training platform, ML compiler, model components libraries, e.t.c
• We are looking for a tech lead to drive the initiatives in distributed training and inference optimization, the key technology to scale our deep learning models to achieve higher performance
• Lead the exploration of the latest technology of distributed training and inference optimization
• Dive deep into the pain points of ML teams and understand model development bottlenecks
• Define and drive the technical strategy for model optimization
• Collaborate with ML teams to ensure smooth adoption of the model optimization solutions
• Promote engineering best practices and cultivate a culture of continuous improvement in the team