Senior Software Engineer - ML Performance

Latitude AI

$179K — $268K *
Consumer Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's degree in Computer Engineering, Computer Science, Electrical Engineering, Robotics or related field with 4+ years of experience (or Master's with 2+ years, or PhD)
  • Expertise in GPU/DSP architecture and heterogeneous embedded computing
  • Fluency in Python, PyTorch, ONNX, numpy, pandas
  • Fluency in modern C++ and data structure optimization
  • Strong understanding of AI deployment on embedded platforms
  • Industry experience in writing performance-critical production code
  • Strong analytical and problem-solving skills

Responsibilities

  • Architect, design, and implement software solutions for large-scale robotics projects
  • Collaborate with teams to integrate and optimize algorithms and models
  • Profile and implement efficiency improvements in code from PyTorch to C++
  • Develop tools to evaluate and optimize AI performance
  • Drive projects end-to-end with a focus on reliability and continuous integration

Benefits

  • High-quality medical, dental, and vision insurance
  • Health savings account with employer match
  • Employer-matched 401(k) retirement plan with immediate vesting
  • Unlimited vacation and 15 paid holidays
  • Paid parental leave and medical leave
  • Daily lunches, snacks, and beverages provided
  • Professional development reimbursement and wellness initiatives
Full Job Description
Meet the team:

In the Performance team, we integrate and optimize software solutions from prototypes to the final product. Our main driver is to make the vehicle react as fast and deterministically as possible to its environment. We are involved in multiple cross-functional projects with stakeholders across the whole company: monitoring performance, optimizing foundation libraries, scheduling accelerators intelligently, and generally squeezing more out of the on-vehicle compute resources.

As part of the Intelligent Systems organization, we are particularly interested in converting photons into a view of the world, primarily via computer vision and machine learning. But our scope is not limited to perception, and influences the whole autonomy stack. We take pride in having a direct impact on the product quality and viability.

Come help us make the world safer, and driving more pleasant.

What you'll do:
  • Architect, design, and implement software solutions that scale from the lab to the customer, for one of the largest-scale robotics projects in history.
  • Collaborate with various teams to understand, integrate and optimize their algorithms and models.
  • Profile and implement efficiency improvements, from PyTorch to C++, balancing requirements with compute constraints.
  • Develop infrastructure and tools which evaluate & optimize AI performance.
  • Drive projects end-to-end and deliver reliable software through continuous integration, automated testing and code reviews, while in a fast-paced, highly autonomous environment which requires a strong safety culture.


What you'll need to succeed:
  • Bachelor's degree in Computer Engineering, Computer Science, Electrical Engineering, Robotics or a related field and 4+ years of relevant experience (or Master's degree and 2+ years of relevant experience, or PhD)
  • Expertise in GPU/DSP architecture and heterogeneous embedded computing.
  • Fluency in Python, PyTorch, ONNX, numpy, pandas.
  • Fluency in modern C++, data structure optimization and parallel programming.
  • Strong understanding of an AI deployment flow on an embedded platform.
  • Industry experience writing production-quality, performance-critical code, and maintaining large codebases.
  • Strong analytical abilities to quickly grasp concepts and design solutions.
  • Strong written and verbal communication skills, to collaborate with stakeholders of various backgrounds.
  • Strong desire to learn and grow, with a curiosity for new domains and technologies, and an R&D mindset.

What will make you stand out:
  • Master's or PhD degree in Robotics, ECE, Computer Science, or a related field.
  • Experience scaling model deployment, profiling or evaluation flows in a production embedded system
  • Experience in optimizing deep learning models through compression, quantization, pruning, kernel fusion, distillation, ... to achieve better inference latency, memory footprint and/or model accuracy.
  • Experience writing SIMD code.
  • Knowledge of perception or autonomous vehicle data flow and software stack.

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
  • 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
  • Adoption/Surrogacy support program
  • 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.

Similar Jobs

More Jobs at Latitude AI

More Consumer Technology Jobs

Find similar Senior Software Engineer - ML Performance jobs: