Senior ML Platform Engineer: Data & Semantic Search

Stack AV; United States; Remote; USD 140,000 - 230,000; Full time

Job summary

Stack AV is advancing autonomous driving with AI-powered data understanding. The ML Data Understanding team builds data mining, curation, and semantic search capabilities to support ML training and evaluation across onboard and offboard systems.

The role emphasizes building scalable infrastructure, utilizing state-of-the-art LLMs, multimodal data handling, and high-throughput data access to accelerate AV product development. Collaboration across data engineering and ML teams is essential.

Qualifications

• 5+ years of relevant domain experience.
• Experience with ML platforms and building ML-based applications.
• Proven track record of building scalable, reliable infrastructure.
• Hands-on background in semantic search infrastructure and embeddings.
• Experience with multi-modal data indexing and model training infrastructure.

Responsibilities

• Architect next-generation multimodal semantic search and data discovery platforms for autonomous driving development.
• Investigate and apply frontier LLMs, visual language models, and RAG techniques.
• Build agentic data mining, semantic search, and data curation workflows in a self-improvement ML loop.
• Develop infrastructure for real-time querying/vector databases and batch/stream processing (Ray, Spark, LanceDB).
• Deliver end-to-end data mining solutions spanning onboard (C++) and offboard (ML & Data Infra).
• Develop end-to-end solution for real-time semantic search (text/images/videos) and vector DBs.
• Identify issues in existing ML infra and proactively improve performance.
• Build low latency/high throughput batch or streaming pipelines.
• Drive technical discussions across orgs and deliver timely solutions.
• Write readable and high-performance Python/C++ code.

Skills

ML platforms

ML applications

Semantic search infra

Vector databases

Distributed data processing

Tools

Ray

Spark

LanceDB

Iceberg

Job description

Stack AV is advancing autonomous driving with AI-powered data understanding. The ML Data Understanding team builds data mining, curation, and semantic search capabilities to support ML training and evaluation across onboard and offboard systems.

The role emphasizes building scalable infrastructure, utilizing state-of-the-art LLMs, multimodal data handling, and high-throughput data access to accelerate AV product development. Collaboration across data engineering and ML teams is essential.