Senior Software Engineer - ML Platform

Latitude AI•Palo Alto, CA

About The Position

Latitude AI (lat.ai) develops automated driving technologies, including L3, for Ford vehicles at scale. We’re driven by the opportunity to reimagine what it’s like to drive and make travel safer, less stressful, and more enjoyable for everyone. When you join the Latitude team, you’ll work alongside leading experts across machine learning and robotics, cloud platforms, mapping, sensors and compute systems, test operations, systems and safety engineering – all dedicated to making a real, positive impact on the driving experience for millions of people. As a Ford Motor Company subsidiary, we operate independently to develop automated driving technology at the speed of a technology startup. Latitude is headquartered in Pittsburgh with engineering centers in Dearborn, Mich., and Palo Alto, Calif. Meet the team: The Machine Learning Platform team is the engine driving all ML initiatives at Latitude. We provide the essential services for every ML pipeline, training run, and critical workflow automation across multiple internal teams. Our flagship product is BlueMind, our in-house platform for orchestrating, executing, and rigorously tracking all ML experiments. Every piece of ML training at Latitude runs on BlueMind, which seamlessly integrates with our high-performance computing cluster, dataset versioning tools, and logging/artifact frameworks. Beyond ML training, we maintain platforms that orchestrate vital business processes, including simulation execution, metrics calculation, bench test execution, map and labeled data production, vehicle/simulation event triage. Our platforms utilize a powerful combination of cutting-edge off-the-shelf and robust in-house web applications and libraries, all integrated with cloud resources. We are looking for experienced candidates who thrive in a fast-paced, collaborative, and distributed environment. If you are a senior engineer who is ready to transform customer needs and innovative ideas into impactful new platform features, we want to hear from you.

Requirements

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)
4+ years of full-stack software engineering experience
Strong communication, problem solving, task prioritization, and teamwork skills
Solid architecture and design skills
Experience coding with Python and familiarity with popular python tools and frameworks such as poetry and fast api
Some knowledge or pytorch, ddp, ray and overall distributed machine learning
Some knowledge of distributed file systems
Experience with SQL databases
Experience with cloud infrastructure (e.g.: AWS, GCP, Terraform, Kubernetes)
Experience with high performance computing environments

Nice To Haves

Experience with workflow automation frameworks (e.g. Airflow, Dagster, Camunda)
Experience with Java and/or other JVM-based languages
Experience with TypeScript and/or JavaScript, Node.js, and frameworks (e.g.: React, Redux, MUI)
Experience with Bazel, Gradle, and/or other build tools

Responsibilities

Drive Innovation on the BlueMind ML Platform: Contribute to the development and enhancement of the in-house BlueMind platform, a critical tool for ML experiment tracking and execution. This platform supports the construction and execution of complex ML training and inference workflows on an on-premise HPC cluster, integrating with other internal and off-the-shelf services for dataset versioning, data lineage, artifact tracking, checkpointing, logging, and metrics tracking
Serve as a Trusted Partner to ML Teams: Regularly engage with and understand the unique needs of diverse ML teams across Latitude (including Scalable Machine Learning, Sensing, Autonomy Behavior, State Estimation, Intelligent Systems Evaluation, Bench Testing, and Virtual Test Ecosystem). You will be instrumental in capturing requirements, authoring robust design documents, implementing high-quality solutions, validating results, and providing critical support for production services
Tackle Complex, High-Impact Challenges: Apply your experience and creativity to solve large-scale, complex technical problems in a dynamic, fast-paced environment, with a direct focus on accelerating Latitude's product launch goals and core mission
Develop and Own Full-Stack Features: Design, implement, and enhance a wide range of full-stack capabilities, spanning web APIs/services, React applications and component libraries, custom and generated client libraries, command-line interfaces, Kubernetes resources, cloud architecture, and crucial integration solutions for our on-premise HPC cluster
Collaborate Across Engineering Boundaries: Work closely with essential supporting teams, including internal Cloud Platform, Product, and Site Reliability teams, as well as external vendors, to ensure the smooth build, deployment, monitoring, and support of our foundational ML platforms

Benefits

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