Stack AV
Stack AV
Stack AV develops AI-powered autonomous trucking systems for the transportation and logistics industries. Its work combines artificial intelligence, machine learning, and cloud technologies to support safer, more reliable, and more efficient trucking operations. The company focuses on improving supply chain intelligence, delivery performance, and business outcomes through autonomous vehicle technology, with safety at the center of its engineering approach. Its multidisciplinary team operates across logistics, manufacturing, and consulting-related areas as it advances intelligent solutions for modern freight transportation.

Staff Software Engineer, ML Training Infrastructure at Stack AV (Remote, Pennsylvania)

Lead the design of scalable machine learning training infrastructure and end-to-end model pipelines. Help advance Stack AV’s autonomous trucking systems through platform engineering, performance optimization, and cross-functional collaboration.

Description

  • Build and refine high-performance training platform components spanning orchestration, training abstractions, control-plane services, observability, and performance optimization.
  • Develop complete machine learning pipelines covering log processing, feature extraction, dataset schemas and storage, model configuration, training, profiling, and acceleration.
  • Evaluate training infrastructure to uncover and address performance constraints.
  • Promote system abstractions and developer tools that help machine learning engineers iterate quickly on models.
  • Adopt open-source technologies that let machine learning engineers independently profile and improve their workflows.
  • Uphold strong engineering standards and help foster a team culture centered on technical excellence.
  • Lead the architecture and implementation of a high-performance, multi-tenant AI training platform.
  • Partner with teams across ML Platform, Infrastructure, Autonomy, and Safety Evaluation.

Requirements

  • Bachelor’s or master’s degree in computer science, engineering, or a related discipline.
  • At least six years of experience developing ML platforms and machine learning applications.
  • Advanced programming ability in Python, C++, or an equivalent language.
  • Experience with Lance, PyTorch, Ray Data, or comparable technologies.
  • Demonstrated success building scalable, dependable infrastructure in a fast-moving environment and partnering with machine learning engineers across modeling teams.
  • Strong grasp of system design tradeoffs, with the communication skills to build alignment among cross-functional teams.
  • Background in model training, model optimization, or large-scale data-processing pipelines.
  • Strong analytical reasoning and problem-solving ability.
  • Clear written and verbal communication skills, including the ability to explain complex technical ideas to nontechnical stakeholders.
  • May need to verify residence, U.S. person status, and/or citizenship to meet U.S. national security and export-control requirements.

Benefits

  • An equal-opportunity workplace focused on inclusion, entrepreneurship, and innovation across gender, race, age, sexual orientation, religion, disability, and identity.

Related Jobs

Snap Inc.

Level 5 Android Software Engineer at Snap Inc. in Los Angeles (Hybrid)

Snap Inc.
5,001 – 10,000 Employees
ConsultingLogisticsMarketing

Build smooth, reliable Snapchat experiences as an Android engineer using Kotlin and Java. Ship mobile features, review code, and apply AI-assisted engineering practices.

Open
General Dynamics Information Technology

VMware Engineer with TS/SCI and Polygraph Clearance

General Dynamics Information Technology

Designs and operates secure VMware virtualization environments for General Dynamics Information Technology’s U.S. government customers. This full-time onsite role supports vSphere infrastructure, systems engineering, automation, and operations in Maryland or Virginia.

Open
Northrop Grumman

Principal Aeronautical Engineer, Vehicle Structures Integration – Melbourne, FL

Northrop Grumman

Lead Vehicle Structures integration initiatives at Northrop Grumman, supporting aerospace processes, tools, training, and technical priorities. Collaborate across Aeronautics Systems disciplines and distributed sites to advance technical excellence and workforce capabilities.

Open