Senior Software Engineer - ML Performance
Latitude AI13 days ago
Base Salary
$179k - $269k/yr
Responsibilities
- Architect, design, and implement scalable software solutions for a large-scale robotics and autonomous driving project.
- Collaborate with teams to integrate and optimize algorithms and machine learning models.
- Profile software and implement efficiency improvements from PyTorch to C++.
- Develop infrastructure and tools to evaluate and optimize AI performance.
- Optimize vehicle compute resources, including foundation libraries and accelerator scheduling.
- Drive projects end-to-end and deliver reliable software through code reviews, continuous integration, and automated testing.
- Maintain a strong safety culture while working with cross-functional stakeholders.
Requirements
- Bachelor’s degree in Computer Engineering, Computer Science, Electrical Engineering, Robotics, or a related field and 4+ years of relevant experience; alternatively, a master’s degree and 2+ years of relevant experience or a PhD.
- Expertise in GPU/DSP architecture and heterogeneous embedded computing.
- Fluency in Python, PyTorch, ONNX, NumPy, pandas, and modern C++.
- Knowledge of data structure optimization, parallel programming, and AI deployment flows on embedded platforms.
- Industry experience writing production-quality, performance-critical code and maintaining large codebases.
- Strong analytical, written communication, verbal communication, learning, and collaboration skills.
- Preferred: master’s or PhD in Robotics, ECE, Computer Science, or a related field.
- Preferred: experience with model deployment, profiling and evaluation flows, deep learning model compression, quantization, pruning, kernel fusion, distillation, SIMD code, and perception or autonomous-vehicle software stacks.
Benefits
- Competitive compensation, annual bonus eligibility, equity compensation, and company benefits.
- Individual and family medical, dental, and vision insurance, plus a health savings account with employer match.
- Employer-matched 401(k) with immediate vesting and employer-paid group term life insurance.
- Paid parental and medical leave, unlimited vacation, and 15 paid holidays.
- Daily lunches, snacks, and beverages at office locations.
- Pre-tax healthcare, dependent-care, and commuter benefits.
- Monthly wellness stipend, adoption/surrogacy support, and backup child and elder care.
- Professional development reimbursement and employee assistance program.
- Discounted legal services, identity theft protection, pet insurance, and other programs.
- Employee resource groups, quarterly team activity stipends, and wellness initiatives.
- Full-time position based in California; U.S. permanent work authorization is required, and visa sponsorship is available.