Serve Robotics is hiring a Software Engineer, ML Infrastructure to help design, build, and maintain the petabyte-scale data and ML platform that powers our robotics fleet. You'll play a pivotal role in enabling data partnerships, ML research, and autonomy engineering by leveraging data from thousands of robots.
What You'll Do
- Develop and maintain highly scalable data processing pipelines for curation, annotation, search, and ML feature extraction.
- Build data discovery and search features, including natural language querying.
- Develop and maintain orchestration and scheduling systems.
- Maintain and evolve data schemas, including a unified data attribute system and scenario tagging.
- Build integrations with annotation providers to review large-scale data preannotations efficiently.
- Collaborate with autonomy engineers to gather feedback, improve documentation, and run tutorials.
What We're Looking For
- BS or MS in computer science with a focus on data engineering or machine learning.
- 3+ years of industry experience building and improving large-volume data processing, feature extraction, and annotation workflows.
- Experience building data mining and search capabilities.
- Proficiency in both Python and SQL is required.
- Solid understanding of data distributions and their impact on ML models.
- Hands-on experience with LLMs, VLMs, embeddings, and vector databases.
- Experience with data annotation providers such as CVAT, LabelBox, or LabelStudio.
Nice to Have
- Experience integrating cloud inference platforms for LLMs/VLMs like ChatGPT or Gemini.
- Experience working with multi-modal data such as Lidar and Camera.
- Experience with robotics systems.
- Experience optimizing large-scale vector databases.
Technical Stack
- Python, SQL
- LLMs, VLMs, Embeddings
- Vector databases
Team & Environment
You will be part of the Machine Learning Infrastructure team, collaborating closely with autonomy and ML teams at Serve Robotics.
Work Mode
This position is open to candidates located in Los Angeles, Miami, Dallas, Atlanta, and Chicago.
Serve Robotics fosters an agile, diverse, and driven culture where we solve complex problems collaboratively and respectfully. We are tech industry veterans pooling our skills to build the future.
