Staff Software Engineer, ML Training Infrastructure
Aurora InnovationAbout the role
Who We Are
Aurora (Nasdaq: AUR) is delivering the benefits of self-driving technology safely, quickly, and broadly to make transportation safer, increasingly accessible, and more reliable and efficient than ever before. The Aurora Driver is a self-driving system designed to operate multiple vehicle types, from freight-hauling semi-trucks to ride-hailing passenger vehicles, and underpins Aurora Horizon and Aurora Connect, its driver-as-a-service products for trucking and ride-hailing. Aurora is working with industry leaders across the transportation ecosystem, including Toyota, FedEx, Volvo Trucks, PACCAR, Uber, Uber Freight, U.S. Xpress, Werner, Covenant, Schneider, and Ryder. For Aurora’s latest news, visit aurora.tech and @aurora_inno on Twitter.
Aurora hires talented people with diverse backgrounds who are ready to help build a transportation ecosystem that will make our roads safer, get crucial goods where they need to go, and make mobility more efficient and accessible for all. We’re searching for a Staff Software Engineer to join Aurora’s Motion Planning Learning team. Our team is responsible for building state-of-the-art Motion Planning machine learning models . The tools we build are mission-critical in allowing our ML modeling teams to explore data, train, and evaluate ML models for our autonomy platform.
In this role, you will
- Develop large scale distributed training models and systems across multiple GPUs
- Collaborate closely with motion planning, systems engineering, and other autonomy groups to define and develop critical ML workflow requirements.
- Define new code designs and publish technical design documents
Required Qualifications
- BS/MS/PhD in Computer Science or related field
- 10+ years of industry experience in full-cycle software development
- Deep experience with large scale distributed systems and training models (preferably with vision and language models)
- Experience with ML or data platforms
- Strong programming skills in Python or C++
- A passion for writing robust, intuitive, high-performance code
- Ability to navigate and work effectively in large codebases
Desirable Qualifications
- Strong programming skills in both C++ and Python
- Machine learning model development experience
- PyTorch, Numpy and Pandas
#LI-JL261
#Mid-Senior
Working at Aurora
At Aurora, we bring together extraordinarily talented and experienced people united by the strength of our values. We operate with integrity, set outrageous goals, and build a culture where we win together — all without any jerks.
We have offices in several locations across the United States, where we encourage team and cross-functional collaboration. Aurora offers competitive medical, dental, and vision benefits, and additional healthcare support including medical transportation reimbursement, fertility, adoption, and surrogacy benefits. We empower our employees and their families with options to further their unique physical, mental, and financial well-being.
Our Learning and Development offerings include Aurora Academy, where our people learn, develop, and practice the essential skills that drive Aurora’s mission, continually up-leveling our team along the way. Our Careers page provides insight into career opportunities at Aurora, and you can find all the latest news on our Blog.
Safety is central to everything we do. Every employee at Aurora has a role in contributing to safety, every step of the way. We seek candidates who take active responsibility, can contribute to building an atmosphere of trust, and invest in the organization's long-term success by working safely — no matter what.
We believe that self-driving technology has broad benefits – including increased access to transportation. To realize those benefits, we need a workfo
Apply for this role
Generate a tailored application kit with a matched cover letter, interview prep, and CV highlights — in under 60 seconds.
Apply Now →Generate Application KitFree account required — sign up in 30s