Senior Software Engineer - ML Infrastructure

Title: Senior Software Engineer - ML Infrastructure Location: Pittsburgh, PA, Palo Alto, CA, Detroit, MI

Meet the team:

The Performance Prediction team builds the data pipelines, machine learning infrastructure, and evaluation systems that help us understand how autonomy behavior evolves from one software release to the next. Our software turns large-scale driving data into clear, trustworthy insights that help teams across Latitude move forward with confidence.

Our work spans the full ML lifecycle, starting with the data. We curate high-quality training and evaluation datasets from real-world and simulated driving, discovering interesting and representative scenarios, labeling them at scale, and organizing them into reusable, well-documented collections. We turn that data into insight, helping engineers understand how the system behaves across a wide variety of driving situations and where the next opportunities lie. We train and ship a range of learned models, from compact classical models to modern deep learning approaches, and build the metrics, experiment tracking, and tools that engineers across autonomy use every day.

This role is a great fit for a software engineer who enjoys building reliable, well-tested Python systems and takes pride in producing ML results that people can trust. Day to day, the work is a mix of data and ML infrastructure, model development and evaluation, and internal tools used by partner teams.

What You’ll Do:

• Curate datasets. Build pipelines that discover, label, and organize driving scenarios from real-world and simulated data into balanced, high-quality training and evaluation sets.
• Build data pipelines. Design and maintain scalable, reproducible data pipelines with versioning, provenance, well-designed train/validation/holdout splits, and clear data contracts.
• Turn data into insight. Explore dataset composition, coverage, and label quality; characterize how behavior evolves across releases and scenario types; and share findings that help shape model, dataset, and autonomy priorities.
• Scale ML infrastructure. Build training, inference, and evaluation pipelines that run smoothly at scale, with robust data loading, checkpointing, and experiment tracking.
• Develop and ship models. Train, tune, and productionize learned models, from classical approaches to deep learning, and integrate them into automated analytics pipelines.
• Own the metrics. Define, implement, and validate model and release metrics such as precision, recall, F1, ROC AUC, calibration quality, and task-specific metrics, with strong statistical rigor.
• Run thoughtful experiments. Set up baselines, ablations, cross-validation, and holdout comparisons; quantify uncertainty; and communicate results that help the team make confident decisions.
• Build tools engineers love. Develop dashboards and services that make it easy to compare software versions, explore results at the scenario level, visualize model outputs, and gather feedback from reviewers.
• Raise the engineering bar. Design clean interfaces between data, modeling, and product layers; champion testing, type checking, code review, and fast, dependable CI.
• Partner across autonomy. Collaborate with autonomy, simulation, data, and infrastructure teams to take ideas from prototype to production.

What You’ll Need To Succeed:

• Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, Robotics, or a related field and 4+ years of relevant industry experience (or Master's degree and 2+ years of relevant experience).
• Strong Python software engineering skills, including designing modular, maintainable, well-tested systems in a shared codebase.
• Experience building end-to-end ML systems, from data pipelines and dataset curation through training, evaluation, and production, using tools such as SQL, pandas, PyTorch, or scikit-learn.
• Experience designing scalable ML infrastructure, such as training and evaluation pipelines, experiment tracking, model serving, or distributed jobs.
• Solid grounding in statistics and experimental design, including choosing evaluation strategies, metrics, and thresholds that support real-world decisions.
• Clear communication and the ability to work effectively across software, ML, and autonomy teams.

Nice to Have:

• Experience with probabilistic forecasting, sequential, or graph-based models, as well as classical methods such as gradient boosting.
• Experience with calibration, uncertainty estimation, ablation studies, error analysis, or release-over-release comparison.
• Experience with data-centric ML, such as active learning, auto-labeling, embedding-based search, or dataset versioning.
• Experience with orchestration, experimentation, and internal tooling such as Dagster, Airflow, Ray, Weights & Biases, Dash, or Streamlit.
• Experience with Bazel or other large-scale build systems.
• Autonomous driving or robotics experience is helpful but not required.

What we offer you:

• Competitive compensation packages
• High-quality individual and family medical, dental, and vision insurance
• Health savings account with available employer match
• Employer-matched 401(k) retirement plan with immediate vesting
• Employer-paid group term life insurance and the option to elect voluntary life insurance
• Paid parental leave and Adoption/Surrogacy support program
• Paid medical leave
• Unlimited vacation and 15 paid holidays
• Daily lunches, snacks, and beverages available in all office locations
• Pre-tax spending accounts for healthcare and dependent care expenses
• Pre-tax commuter benefits
• Monthly wellness stipend
• Backup child and elder care program
• Professional development reimbursement
• Employee assistance program
• Discounted programs that include legal services, identity theft protection, pet insurance, and more
• Company and team bonding outlets: employee resource groups, quarterly team activity stipend, and wellness initiatives

Learn more about Latitude’s team, mission and career opportunities at lat.ai!
The expected base salary range for this full-time position in California is $179,200 - $268,800 USD. Actual starting pay will be based on job-related factors, including exact work location, experience, relevant training and education, and skill level. Latitude employees are also eligible to participate in Latitude’s annual bonus programs, equity compensation, and generous Company benefits program, subject to eligibility requirements.

Candidates for positions with Latitude AI must be legally authorized to work in the United States on a permanent basis. Verification of employment eligibility will be required at the time of hire. Visa sponsorship is available for this position.