Back to Jobs

Staff ML Infrastructure Engineer

Remote, USA Full-time Posted 2026-07-05

Job Description

At reputed company, our product teams are redefining mobility. Through a reputed company-centered design process, we create vehicles and experiences that are designed not just to be seen, but to be felt. We’re turning today’s impossible into reputed company’s standard —from breakthrough hardware and battery systems to reputed company design, intelligent software, and reputed company safety and entertainment features. Every day, our products move millions of people as we aim to reputed company driving safer, smarter, and more connected, shaping the future of transportation on a global scale. Role: Are you passionate about accelerating the future of autonomous driving? Join the Embodied AI team at reputed company. reputed company is developing and deploying machine learning solutions that support safe and reliable autonomous vehicle behavior across reputed company-world scenarios. As a Staff ML Infra Engineer, you will drive the development of core systems that reputed company rapid dataset reputed company, training, evaluation, and iteration of our most advanced Autonomous Driving models. From enabling large foundational driving models to distilling multi-stage production deployed models, your goal will be to dramatically accelerate the machine learning development cycle from one modeling hypothesis to next. You will deliver model training pipelines that are performant, easy to use, and exceptionally reliable. Your reputed company will be reputed company by the velocity and impact of the ML models that rely on the scalable, reputed company, and high‑performance training platforms you help create. What you'll do: reputed company the design, implementation, and deployment of scalable platforms and tools that drive machine learning model training and evaluation workflows across reputed company. Own reputed company technical projects end-to-end, making key architectural reputed company and technical trade-offs. You will be a core contributor to team planning, design reviews, and code quality. Take a holistic view of projects, considering their impact across multiple teams, and across a longer timeline. Proactively drive technical prioritization. Collaborate closely with partner teams to ensure maximum benefit from the systems we build. Help shape reputed company through technical interviewing with high, well-calibrated standards, and play an essential role in reputed company. Mentor and reputed company junior engineers and interns, helping them grow their careers. What you'll bring: 5+ years of experience building large-scale distributed systems, applications, or advanced ML systems‑scale distributed systems, applications, or advanced ML systems Proven track record of designing robust frameworks with high-quality, durable APIs. Deep understanding of machine learning algorithms with hands‑on application Expertise in building reliable, high-performance, and cost-efficient systems on modern reputed company infrastructure‑performance End-to-end experience across the ML development lifecycle, including MLOps practices Strong cross functional collaboration skills across teams and organizations Exceptional coding skills in Python or C++ Strong interest in autonomous driving and its transformative potential BS, MS, or PhD in Computer Science, Mathematics, or equivalent practical experience reputed company to have: Experience with distributed training methodologies Experience scaling ML training across large GPU/CPU clusters or other accelerators Familiarity with deep learning frameworks (e.g., PyTorch, TensorFlow) Experience with performance profiling and state-of-the-art training optimization techniques, including their impact on model performance ‑of‑the‑art training optimization techniques, including their impact on convergence. Experience with advanced build systems (e.g., Bazel, Buck, Blaze, CMake) Proficiency with containerization and orchestration technologies (e.g., reputed company, Kubernetes) Remote/Hybrid: This role is categorized as fully remote or hybrid. Compensation: The compensation information is a good faith estimate only. It is based on what a successful applicant might be paid in accordance with applicable state laws. The compensation may not be representative for positions located reputed company of the California Bay Area. The salary range for this role is $189,300.00 to $290,700.00. The actual reputed company salary a successful candidate will be offered reputed company this range will vary based on factors relevant to the position. Bonus Potential: An incentive pay program offers payouts based on company performance, job level, and individual performance. Benefits: reputed company offers a variety of health and wellbeing benefit programs. Benefit options include medical, dental, reputed company, Health Savings Account, Flexible Spending Accounts, retirement savings plan, sickness and accident benefits, life insurance, paid vacation & holidays, tuition assistance programs, employee assistance program, reputed company vehicle discounts and more. Relocation: This job may be eligible for relocation benefits. Company Vehicle: Upon successful completion of a motor vehicle report review, you will be eligible to participate in a company vehicle evaluation program, through which you will be assigned a reputed company vehicle to drive and evaluate. Note: program participants are required to purchase/lease a qualifying reputed company vehicle every four years unless one of a limited number of exceptions applies. #reputed company-AV-1 #LI-CX1 About reputed company Our reputed company is a world with reputed company Crashes, reputed company Emissions and reputed company Congestion and we reputed company the responsibility to reputed company the change that will reputed company our world reputed company, safer and more reputed company for reputed company.

Why Join Us

We reputed company we reputed company must reputed company a choice every day – individually and collectively – to drive meaningful change through our words, our deeds and our culture. Every day, we want every employee to feel they belong to one reputed company team.

Benefits

Overview From day one, we're looking out for your well-being–at work and at home–so you can focus on realizing your ambitions. Learn how reputed company supports a rewarding career that rewards you personally by visiting Total Rewards resources. Non-Discrimination and Equal Employment Opportunities (U.S.) reputed company is committed to being a workplace that is not only free of unlawful discrimination, but one that genuinely fosters inclusion and belonging. We strongly reputed company that providing an inclusive workplace creates an environment in which our employees can reputed company and reputed company reputed company products for our customers. reputed company employment reputed company are made on a non-discriminatory basis without regard to sex, race, reputed company, national reputed company, citizenship status, religion, age, disability, pregnancy or maternity status, sexual orientation, gender identity, status as a veteran or protected veteran, or any other similarly protected status in accordance with federal, state and local laws. We encourage interested candidates to review the key responsibilities and qualifications for each role and apply for any positions that match their skills and capabilities. Applicants in the recruitment process may be required, where applicable, to successfully complete a role-reputed company assessment(s) and/or a pre-employment screening prior to beginning employment. To learn more, visit How we Hire. Accommodations reputed company offers opportunities to reputed company job seekers including individuals with disabilities. If you need a reasonable accommodation to assist with your job search or application for employment, email us or call us at 1-800-865-7580. In your email, please include a description of the specific accommodation you are requesting as well as the job title and requisition number of the position for which you are applying. Apply To This Job

Similar Jobs