logo inner

Staff Software Engineer, ML Platform

Stack AVPittsburgh, Pennsylvania, United StatesRemote, Onsite

About Stack:


Stack is developing revolutionary AI and advanced autonomous systems designed to enhance safety, reliability, and efficiency of modern operations. Stack's autonomous technology incorporates cutting-edge advancements in artificial intelligence, robotics, machine learning, and cloud technologies, empowering us to create innovative solutions that address the needs and challenges of the dynamic trucking transportation industry. With decades of experience creating and deploying real world systems for demanding environments, the Stack team is dedicated to developing an autonomous solution ecosystem tailored to the trucking industry's unique demands.

About the Team:


In the ML Training and DevEx team, our mission is to provide a reliable, scalable, and easy to use training framework for modeling needs of Stack AV. In addition, this team is responsible for the overall developer experience of ML engineers including building tools for testing, validation, and understanding models and the data used to train them. Finally, this team is responsible for model optimization and deployment.

What Success Looks Like:


  • Experience with both ML Platforms and building ML-based applications (bonus point if you have modeling experience).
  • Experience building scalable, reliable infra at a fast-paced environment.
  • Ability to work across teams.
  • Experience building or using ML infra built for a large number of customer teams.
  • A deep understanding of design tradeoffs and ability to articulate those tradeoffs and work with others on getting alignment.
  • Experience with building ML models or ML infra in the domains of autonomous vehicles, perception, and decision making (desirable but not required).
  • Experience with model training, model optimization, or large data processing pipelines.

Preferred Experience:


  • Knows how to push the GPU to its limit from Python to CUDA kernel level.
  • Built the inference or training loop for a large model (ideally with LLM flavor).
  • Shipped ML products (NLP, computer vision, recommender systems, etc.) at scale to make business impact
  • Knows how to build low latency / high throughput batch or stream processing pipelines.
  • Knows how to write (readable) high performance C++.
  • Prior AV experience.

#LI-AW1We 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.

Life at Stack AV

Stack is revolutionizing the transportation industry through the deployment of advanced autonomous systems designed to meet the safety, reliability, and efficiency demands of the trucking industry.
Thrive Here & What We Value- Emphasis on innovation and entrepreneurship- Commitment to building an inclusive culture across various aspects of diversity- Focus on creating better supply chains and improving business outcomes- Compliance with U.S. national security laws and regulations- Support for diverse teams producing the best ideas and outcomes- Equal opportunity employment- Strong focus on safety engineering and autonomous vehicle development- Willingness to consider candidates for alternative positions if necessary- Inclusive workplace- Commitment to building a culture of innovation, entrepreneurship, and diversity across various aspects of identity
Your tracker settings

We use cookies and similar methods to recognize visitors and remember their preferences. We also use them to measure ad campaign effectiveness, target ads and analyze site traffic. To learn more about these methods, including how to disable them, view our Cookie Policy or Privacy Policy.

By tapping `Accept`, you consent to the use of these methods by us and third parties. You can always change your tracker preferences by visiting our Cookie Policy.

logo innerThatStartupJob
Discover the best startup and their job positions, all in one place.
Copyright © 2024