Stack is developing revolutionary AI and advanced autonomous systems designed to enhance safety, reliability, and efficiency of modern operations in the trucking transportation industry. The Foundational Models team aims to leverage recent advances in foundational models and generative AI to improve the safety and efficiency of the Autonomy Stack and co-design a new HW/SW platform to run these models and workflows.
Requirements
- Experience in contributing to or leveraging one or more of the recent advances related to foundational models such as pre-training models, RL based fine-tuning , neural rendering, diffusion based synthetic image/video generation, World Foundational Models.
- Experience with processing large datasets, training ML models in distributed environments.
- Track record of developing end to end pipelines for training Machine Learning Models
- Strong experience in Python and at least one of popular Deep Learning frameworks like Pytorch, Tensorflow, JAX.
- Experience with containerization, FastAPI, REST /GraphQL, SQL.
- Experience with autonomous driving and/or robotics.
Responsibilities
- Contribute to the technical direction of the team and broader perception org on leveraging foundational models and related techniques.
- Design and implement offboard pipelines to train, distill, evaluate and serve Machine Learning Models.
- Demonstrate that these pipelines work with open source data and contribute to open-source libraries.
- Proactively identify opportunities for improvements and work with leadership team on prioritization of resolutions.
- Align stakeholders with strong presentation and communication skills.
- Lead challenging technical topics and build consensus.
Other
- BS (with 6 years of experience), MS (with 5 years of experience) or PhD (with 3 years of experience) in developing machine learning models or systems.
- Track record of driving projects from start to completion, including conception, problem definition, experimentation, iteration, and finally productization.
- 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.
- 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.
- This position may be contingent upon Stack AV verifying a candidate’s residence, U.S. person status, and/or citizenship status.
