Senior ML Platform Engineer - Multimodal Search & Data

Stack AV; Pittsburgh (Allegheny County); Hybrid; USD 140,000 - 210,000; Full time

Job summary

Stack AV is advancing autonomous driving through AI-powered data understanding and multimodal semantic search. The team focuses on data mining, labeling, and scalable infrastructure to accelerate ML workloads across onboard C++ and offboard ML infra. You will work with Ray, Spark, LanceDB, and Iceberg to build robust data platforms for AV development.

Join a fast-paced environment shaping the data stack for autonomous systems, with emphasis on performance, scalability, and cutting-edge AI models.

Qualifications

• Experience with ML platforms and building ML-based applications.
• 5+ years of relevant domain experience.
• Proven track record of building scalable, reliable infrastructure in a fast-paced environment.
• Deep understanding of design trade-offs with the ability to articulate those trade-offs and achieve alignment with others.
• Hands-on background in developing semantic search infrastructure, generating multimodal embeddings, and managing vector databases.

Responsibilities

• Architect next-generation multimodal semantic search and data discovery platforms for autonomous driving development.
• Investigate and apply frontier large language models and state-of-the-art RAG techniques.
• Build out data mining, semantic search, and data curation workflows as part of a recursive self-improvement ML loop.
• Develop data understanding platform infrastructure for real-time querying/vector databases and batch/stream processing using Ray, Spark, LanceDB, or similar.
• Deliver end-to-end data mining solutions spanning onboard (C++) and offboard (ML & Data Infra) infrastructure to accelerate AV product development.
• Develop end-to-end semantic search services (text/images/videos) and vector DBs.
• Identify and improve ML infra performance and build low latency/high throughput processing pipelines.
• Drive technical discussions across orgs and deliver timely solutions.
• Write readable and high-performance Python/C++ code.

Skills

ML platforms

Semantic search

Vector databases

Python/C++

Autonomous vehicles

Tools

Ray

Spark

LanceDB

Iceberg

Job description

Stack AV is advancing autonomous driving through AI-powered data understanding and multimodal semantic search. The team focuses on data mining, labeling, and scalable infrastructure to accelerate ML workloads across onboard C++ and offboard ML infra. You will work with Ray, Spark, LanceDB, and Iceberg to build robust data platforms for AV development.

Join a fast-paced environment shaping the data stack for autonomous systems, with emphasis on performance, scalability, and cutting-edge AI models.