Logo
Nuro

Software Engineer, ML Infra

Nuro, Mountain View, California, us, 94039


Who We Are

Nuro exists to better everyday life through robotics. Founded in 2016, Nuro is a leading autonomous technology company with vehicles on road today in California and Texas. The company’s core technology is the Nuro Driver, an integrated autonomous driving system consisting of state-of-the-art, AI-first software and custom-built sense and compute hardware. The first commercial application of the Nuro Driver is autonomous goods delivery with our custom, electric, zero-occupant vehicles in partnership with some of the most recognized brands in the world including Uber and FedEx.About The Role

The ML Infra team is growing and we are looking for talented engineers to join us. Nuro is pursuing an ML-first software stack. 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, etc.We are looking for engineers to drive the optimization solutions for training and inference, the key technology to scale our deep learning models to achieve higher performance. Your role combines the exploration of leading-edge optimization frameworks, engaging with modern hardware functionalities, and researching performance-optimized deep learning architectures. Leveraging this knowledge, you will engineer products with high engineering quality to optimize the deep learning models across the entire company.About The WorkDistributed training speed up and cost reduction.Distributed training platform development.Profiling performance of onboard inference and developing optimizations.ML compiler architecture and usability.About You2+ years of work or research experience in machine learning, ML Infra, 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.Bonus PointsExperience with CUDA, Cublas, Cudnn or any other Nvidia SDKs.Experience with model quantization.Experience with compilers or ML compilers (e.g. TensorRT, TVM, XLA, Clang).Experience with AI algorithms and hardware codesign (e.g. Depthwise Conv, Flash Attention).Experience with distributed training speedup (e.g. Horovod, DeepSpeed).Experience with deep learning algorithms (e.g. YOLO, Transformer).At Nuro, your base pay is one part of your total compensation package. For this position, the reasonably expected base pay range is between $138,225 and $207,575 for the level at which this job has been scoped. Your base pay will depend on several factors, including your experience, qualifications, education, location, and skills. In the event that you are considered for a different level, a higher or lower pay range would apply. This position is also eligible for an annual performance bonus, equity, and a competitive benefits package.At Nuro, we celebrate differences and are committed to a diverse workplace that fosters inclusion and psychological safety for all employees. Nuro is proud to be an equal opportunity employer and expressly prohibits any form of workplace discrimination based on race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, veteran status, or any other legally protected characteristics.

#J-18808-Ljbffr