Senior Software Engineer - Performance Prediction

About the role

The Performance Prediction team at Latitude AI builds machine learning models and evaluation pipelines to understand how autonomy behavior changes across software releases. This role involves building production software for model training, offline evaluation, and release-comparison workflows, developing and improving learned models, and creating internal tools to support partner teams.

Responsibilities

• Build production software for model training, offline evaluation, and release-comparison workflows
• Develop and improve learned models for performance prediction, including behavior classifiers and probabilistic prediction models
• Design training, validation, and holdout strategies that produce trustworthy results
• Define and track model and release metrics such as precision, recall, F1, ROC AUC, calibration quality, and task-specific forecasting metrics
• Run experiments, tune models, and analyze results with strong statistical rigor
• Build internal tools that help engineers compare software versions, inspect model outputs, and investigate regressions
• Partner with autonomy, simulation, and infrastructure teams to move ideas from prototype to production
• Raise engineering quality through testing, code review, CI, and maintainable interfaces across data, modeling, and product layers

Requirements

• Bachelor's degree in Computer Engineering, Computer Science, Electrical Engineering, Robotics or a related field and 4+ years of relevant experience
• Strong software engineering skills in Python, including experience building modular, maintainable, well-tested systems in a shared codebase
• Experience developing, training, tuning, or productionizing supervised ML models
• Strong grounding in statistics and experimental design, including experience designing model training and evaluation tests
• Experience selecting and interpreting model metrics, thresholds, and tradeoffs for real-world decision-making
• Experience with ML tooling such as PyTorch, scikit-learn, or similar frameworks
• Experience working with large datasets using SQL, pandas, and NumPy
• Strong communication skills and the ability to work effectively across software, ML, and autonomy teams

Qualifications

• PhD in Computer Science, Machine Learning, Statistics, Robotics, or a closely related field is preferred
• Equivalent research-heavy industry experience is highly valued
• Experience with probabilistic forecasting, trajectory prediction, sequential modeling, or graph-based models
• Experience with classical ML methods such as random forests, gradient boosting, or calibrated linear models
• Experience with calibration, uncertainty estimation, ablation studies, error analysis, or release regression methodology
• Experience building internal analytics or ML tools with Dash, Plotly, Streamlit, or similar frameworks
• Experience with workflow orchestration or experimentation tools such as Dagster, Airflow, or Weights and Biases
• Experience with Bazel or other large-scale build systems
• Prior autonomous driving experience is helpful but not required

Skills

• 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
• Paid medical leave
• Unlimited vacation
• 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
• 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

Benefits

• 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.
• 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.
• We are an Equal Opportunity Employer committed to a culturally diverse workforce.