Staff Software Engineer, ML Developer Velocity

About

With customers at its core, Stack AV is focused on revolutionizing the way businesses transport goods, designing solutions to alleviate long-standing issues that have plagued the trucking industry including driver shortages, lagging efficiency in uptime per vehicle, overarching safety concerns, high operating costs, and elevated emission levels. By building safe and efficient autonomous trucking solutions, Stack AV is creating better and smarter supply chains for its partners, improving business outcomes for its customers, delivering goods to end-users faster, and ultimately moving the trucking industry forward.

About The Team

The ML Platform org at Stack AV develops the platform that powers all AI at Stack, including data collection, model training, testing, and deployment on the autonomous vehicles. The ML Developer Velocity team is responsible for the overall velocity of ML engineers, building the tools to enable development of all the AI applications that make autonomous vehicles a reality as safely and as efficiently as possible. The scope for this team includes building tools for experiment management, model testing & safety validation, and training data influence analysis.

Position Overview

We are seeking an experienced and hands-on engineer for our ML Developer Velocity team. The ideal candidate has a strong infrastructure and machine learning background, with some previous experience either as an ML Engineer or having worked closely with ML Engineers.

Team Responsibilities

Work regularly with ML Engineers to understand their development workflow and improve it. This includes understanding the overall flow of going from on-road improvement requirements to having a deployed model on the truck and improving the workflow.Automation of manual work is a large part of this job. Examples include converting manual work to automated pipelines (on Flyte) and creating metrics that can make automated go/no-go decisions on model deployment.Experiment management using third-part tools (like MLFlow) and connecting this to the rest of development workflow (data infra and training infra).Work with open source teams to merge our changes upstream.

Key Responsibilities

The successful candidate will have a significant impact on what models can be deployed on the AV hardware as well as the development experience and velocity of ML engineers across the company.

Understand the needs and pain points of ML Engineers across the company and develop solutions to improve the overall velocity and user experience.Propose solutions using a mixture of front end and backend changes to meet our customer needs.Collaboration: Collaborate with cross-functional teams to understand data requirements and design appropriate solutions.Technology Stack: Stay updated with the latest technologies and trends in ML. Promote Engineering Excellence: Set a culture of engineering excellence within the team and work closely with the management and customer teams to balance between speed of delivery and quality of engineering artifacts.

Qualifications

Education: Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field.Experience: 5+ years of experience in ML Infra / MLE. Familiarity with infra / front end ideal.Technical Skills:Strong programming skills in Python.Strong service system design skills.Comfortable driving designs across multiple groups to unblock a project.Familiar with ML Engineer user journeys and comfortable working closely with ML Engineers.Comfortable working on all parts of the stack, front end, backend, generic infra, and ML infra.(preferred) prior experience with offline ML pipelines and experiment management tooling. In particular, Flyte, MLFlow and Weights and Biases.Problem-Solving: Strong analytical and problem-solving skills.Communication: Excellent verbal and written communication skills, with the ability to convey complex technical concepts to non-technical stakeholders.

We are proud to be an equal opportunity workplace. We believe that diverse teams produce the best ideas and outcomes. We are committed to building a culture of inclusion, entrepreneurship, and innovation across gender, race, age, sexual orientation, religion, disability, and identity.

Check out our Privacy Policy.

Please Note: Pursuant to its business activities and use of technology, Stack AV complies with all applicable U.S. national security laws, regulations, and administrative requirements, which can restrict Stack AV’s ability to employ certain persons in certain positions pursuant to a range of national security-related requirements. As such, this position may be contingent upon Stack AV verifying a candidate’s residence, U.S. person status, and/or citizenship status. This position may also involve working with software and technologies subject to U.S. export control regulations. Under these regulations, it may be necessary for Stack AV to obtain a U.S. government export license prior to releasing its technologies to certain persons. If Stack AV determines that a candidate’s residence, U.S. person status, and/or citizenship status will require a license, prohibit the candidate from working in this position, or otherwise be subject to national security-related restrictions, Stack AV expressly reserves the right to either consider the candidate for a different position that is not subject to such restrictions, on whatever terms and conditions Stack AV shall establish in its sole discretion, or, in the alternative, decline to move forward with the candidate’s application.

Key Skills

• Python