TL;DR

Principal Software Engineer (AI): Shaping and building next-generation ML Infrastructure and significantly reduce the time to develop and deploy large-scale ML and Foundational models to our robotaxi with an accent on data, compute, model training, and serving infrastructure. Focus on enabling innovation in autonomous driving and enhancing our rider experience.

Location: Hybrid

Salary: $373,000-$469,000 a year

Company

%hirify_global% is on a mission to reimagine transportation and ground-up build autonomous robotaxis that are safe, reliable, clean, and enjoyable for everyone.

What you will do

Develop and execute a strategic vision for ML Infrastructure that will unlock innovation in autonomous driving and enhance our rider experience.
Lead the design and implementation of infrastructure spanning all stages of an ML lifecycle from data preparation to training to evaluation, deployment, and serving.
Collaborate closely with cross-functional teams, including ML researchers, software engineers, data engineers, and hardware engineers, to define requirements and align on architectural decisions.
Enable the engineers in the team to grow their careers by providing technical guidance and mentorship.

Requirements

Experience building and managing large-scale ML infrastructure that powers the development of large-scale ML models
Excellent leadership skills with a demonstrated ability to lead high-performing engineering teams.
Strong experience with training frameworks like PyTorch, JAX, etc., leveraging GPUs efficiently for distributed model training.
Experience with GPU-accelerated inference using TensorRT, Ray Serve, or similar frameworks.
Proficient in Python and/or C++.

Nice to have

Experience enabling the development and deployment of large-scale Foundation models.
Experience working on large-scale data infrastructure and big data processing frameworks like Apache Spark.
Experience working in the AV domain supporting Perception, Prediction, Planner et al.

Culture & Benefits

Comprehensive package of benefits including paid time off (e.g. sick leave, vacation, bereavement), unpaid time off.
Stock Appreciation Rights, Amazon RSUs.
Health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.