Senior Staff Software Engineer, Stack AV
ML Acceleration Lead
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
- The ideal candidate will have a deep understanding of GPUs and optimization, excellent leadership skills, and the ability to drive technical excellence
- Education: Bachelor’s or Master’s degree in Computer Science, Engineering, or a related field
- Experience: 5+ years of experience in GPU programming and optimization
- Technical Skills:
- 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
- Leadership Skills: Proven ability to lead and mentor a team, manage projects, and drive technical initiatives
- 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, visionary, and hands-on technical lead for our ML acceleration team
- This role will be responsible for designing the architecture and leading a team to automate the optimization and deployment on complex ML models (including transformer-based models such as VLM models) for all the next-gen AI Autonomous Vehicle applications in the company
- 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
- Implement optimizations using CUDA, Triton, and custom kernels
- Collaborate with ML researchers to balance model accuracy and speed
- Tech Leadership: Lead efforts within the team as well as cross-team projects related to model optimization and deployment
- 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: 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
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