Staff Software Engineer, Stack AV
Machine Learning Acceleration, Individual Contributor
Autonomous systems for the transportation industry
Open for applications
Autonomous systems for the transportation industry
201-500 employees
Open for applications
201-500 employees
Company mission
To revolutionize the transportation industry through the deployment of advanced autonomous systems.
Role
Who you are
- We are seeking passionate individuals to join our team, driven to develop and deploy advanced AI products that redefine industry standards, making processes safer, faster, and more efficient
- If you thrive in fast-paced tech environments and are eager to contribute to transformative AI solutions, we invite you to join us in shaping the future of technology
- Education: Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field
- Experience: 5+ years of experience (including experience with GPU programming and optimization)
- Strong programming skills in C++ and Python
- Proven experience in GPU programming and optimization
- Familiarity with deep learning frameworks, especially PyTorch
- CUDA programming
- Triton language for GPU kernels
- PyTorch optimization techniques
- TensorRT implementation
- ONNX model conversion and deployment
- Custom GPU kernel development
- Deep understanding of GPU architectures and performance optimization
- 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
What the job involves
- The training and deployment team, part of the ML Platform org at Stack AV, is responsible for the platform that helps the AI team to build models, optimize, test, and deploy them on the autonomous vehicles
- We are seeking an experienced and hands-on engineer for our ML acceleration team. The ideal candidate will have a deep understanding of GPUs and optimization, excellent collaboration skills, and the ability to drive technical excellence
- Analyze and profile ML models to identify performance bottlenecks
- Use OSS tooling to enhance our platform to enable ML engineers to profile models and optimize them (e.g., through quantization)
- Automate the process of exporting the model to optimized format (e.g., TensorRT) and deploying them. This including transformer-based models such as VLM models
- Implement optimizations using CUDA, Triton, and custom kernels
- Collaborate with ML researchers to balance model accuracy and speed
- Develop and implement efficient model export, optimization, and profiling solutions to enhance performance and streamline deployment of machine learning models across various hardware platforms
- 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 inference and ML accelerators
- Performance Optimization: Identify and resolve performance bottlenecks in models
- Promote Engineering Excellence: Maintain a high bar for engineering excellence in their own work but also set a culture of engineering excellence within the team
Our take
Driver shortages, decelerating efficiency and broad safety concerns have plagued the US trucking industry for some time now. High operating costs paired with increasingly scrutinized high emissions are leading to concern over the future of the logistics and goods transport market. Stack AV’s autonomous trucking solution aims to lighten the load of these issues, using AI to create smart supply chains and autonomous driving systems.
Its founding team have long been exploring the possibilities of autonomous transport through combined experience with Uber, and Google smart cars and shared founding roles at Argo AI, a similar company that disbanded in 2022 after it failed to become profitable for shareholders. Hopefully, however, the launch of Stack AV will balance the decades of experience as well as hard lessons learnt from the failed predecessor company and will find new ways to make autonomous driving tech a sustainable investment.
With a staggering $1 billion invested in Stack AV by SoftBank in 2023, the company is well-armoured for the time being. Its main competitor is Aurora, which underwent a successful IPO in 2021 but never achieved the high level of early funding that Stack AV did. It’s a promising start then for the Argo AI veterans' new venture.
Kirsty
Company Specialist at Welcome to the Jungle
Company
Funding (1 round)
Sep 2023
$1bn
SERIES A
Total funding: $1bn
Company values
- SAFETY - Safety is our top priority. Individually and collectively, we embrace our obligation to be responsible and ensure that safety is our company's bedrock
- HUMILITY - We operate with respect for all, and are humble in our achievements and mistakes. We take ownership of our conduct, embrace one another's differences, and uphold an environment of kindness and dignity
- We operate with respect for all, and are humble in our achievements and mistakes. We take ownership of our conduct, embrace one another's differences, and uphold an environment of kindness and dignity
- GRIT - The path to success is rarely straight or smooth. We succeed as a team through disciplined execution, working together to understand the problems at a fundamental level and developing thoughtful and practical solutions
- EXCELLENCE - As a team, we execute and deliver a world class product that brings value to our customers by solving their critical business needs in a safe and efficient manner. Our accomplishments together are greater than any effort individually
- TRANSPARENCY - We are transparent and honest about the facts of our execution and hold ourselves accountable to drive real, continuous improvement through our work and to foster public trust and support of our products and services
- PASSION - We are passionate about our mission, and our culture. Our passion motivates our success and creates an environment where our teams thrive
Company HQ
Downtown, Pittsburgh, PA
Articles
Leadership
Brian Salesky
(CEO)Previously CEO of Argo AI, Director of Hardware Development for self-driving cars at Google, Principal Commercialization Specialist at the National Robotics Engineering Center, and Technical Program Manager at Google.
Peter Rander
(President)Previously President and Co-Founder at Argo AI, Engineering Lead at Uber and Principal Commercialization Specialist at the National Robotics Engineering Center.
Brett Browning
(CTO)Previously CTO at Argo AI, Senior Engineering Manager at Uber Advanced Technologies Center, Co-Founder at Mine Vision Systems and a Senior Systems Scientist at Carnegie Mellon University.
Share this job
View 9 more jobs at Stack AV