Staff Software Engineer - ML Training

March 5

Apply Now
Logo of Stack AV

Stack AV

Stack AV is a company that is revolutionizing the transportation industry through its autonomous trucking solutions, driven by advanced artificial intelligence. The company focuses on developing AI-powered autonomous systems to enhance safety, reliability, and efficiency in trucking operations. Stack AV is committed to addressing the challenges of the trucking industry by designing smart solutions to improve supply chain intelligence, business outcomes, and delivery speed. Safety is a core principle, and the company leverages cutting-edge AI, machine learning, and cloud technologies to innovate within the industry.

πŸ“‹ Description

β€’ 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. β€’ The ML Training Team’s core mandate is training models as fast as possible for the company. β€’ The team’s main focus is ensuring our models have 100% gpu utilization and can scale linearly from 8 gpus -> 256 gpus. β€’ We also invest in tooling to empower our MLEs, by building profiling/debugging tools, setting up efficiency monitoring and integrating our trainer into our experiment management system.

🎯 Requirements

β€’ Experience: 5+ years as a SWE, ideally building infrastructure/customer facing product, experience in AV or robotics is also great. β€’ Ideal Skills: β€’ 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. β€’ 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. β€’ Machine Learning Expertise is preferred but not necessary. β€’ 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. β€’ Desired Attributes: β€’ High customer empathy, able to communicate with customers well β€’ Comfortable reading papers / keeping up with SOTA ML literature

Apply Now

Discover 100,000+ Remote Jobs!

Join now to unlock all jobs

Discover hidden jobs

We scan the internet everyday and find jobs not posted on LinkedIn or other job boards.

Head start against the competition

We find jobs within 24 hours of being posted, so you can apply before everyone else.

Be the first to know

Daily emails with new job openings straight to your inbox.

Choose your membership

Cancel anytime

Loved by 10,000+ remote workers

Wall of Love

Frequently asked questions

We use powerful scraping tech to scan the internet for thousands of remote jobs daily. It operates 24/7 and costs us to operate, so we charge for access to keep the site running.

Of course! You can cancel your subscription at any time with no hidden fees or penalties. Once canceled, you’ll still have access until the end of your current billing period.

Other job boards only have jobs from companies that pay to post. This means that you miss out on jobs from companies that don't want to pay. On the other hand, Remote Rocketship scrapes the internet for jobs and doesn't accept payments from companies. This means we have thousands more jobs!

New jobs are constantly being posted. We check each company website every day to ensure we have the most up-to-date job listings.

Yes! We’re always looking to expand our listings and appreciate any suggestions from our community. Just send an email to Lior@remoterocketship.com. I read every request.

Remote Rocketship is a solo project by me, Lior Neu-ner. I built this website for my wife when she was looking for a job! She was having a hard time finding remote jobs, so I decided to build her a tool that would search the internet for her.

Why I created Remote Rocketship

Choose your membership

Cancel anytime

Loved by 10,000+ remote workers
Built by Lior Neu-ner. I'd love to hear your feedback β€” Get in touch via DM or lior@remoterocketship.com