Software Engineer, ML Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

Serve Robotics · 2 months ago

Software Engineer, ML Infrastructure

Serve Robotics is reimagining urban deliveries with their sidewalk robot, aiming to enhance efficiency and accessibility in city logistics. The Software Engineer on the ML Infrastructure team will design, build, and maintain a data and ML platform, collaborating with various teams to optimize data processing and discovery features.

Artificial Intelligence (AI)Food DeliveryLogisticsRobotics
check
H1B Sponsor Likelynote

Responsibilities

Develop and maintain highly scalable data processing pipelines for data curation, annotation, search and ml feature extraction
Build data discovery features for the platform
Create and maintain search features such as natural language querying
Develop and maintain our orchestration and scheduling systems
Maintain and evolve our data schemas such as unified data attribute system, scenario tagging and management
Build integrations with annotation providers to efficiently review large scale data preannotations
Collaborate with autonomy engineers to collect feedback, improve documentation, and run tutorials on platform features

Qualification

Data engineeringMachine learningPythonSQLData processing pipelinesData miningData annotation workflowsData distributionsLLMsVLMsCloud inference platformsMulti Modal dataRobotics systemsVector databases

Required

BS or MS in computer science with focus in data engineering and/or machine learning
3+ years of industry experience building, running and improving large-volume data processing, feature extraction, data annotation workflows
Experience building data mining and search capabilities
Experience with both Python and SQL is required
Solid understanding of data distributions and their impact on ML Models
Hands-on experience and good understanding of LLMs, VLMs, embeddings, vector databases
Experience with data annotation providers such as CVAT, LabelBox, LabelStudio, etc

Preferred

Experience with integrating cloud inference platforms for LLMs/VLMS (ChatGPT, Gemini, etc)
Experience working with Multi Modal data (Lidar, Camera, etc)
Experience with robotics systems
Experience optimizing large scale vector databases

Company

Serve Robotics

twittertwittertwitter
company-logo
Serve Robotics is an autonomous robotic delivery company that develops AI-powered sidewalk delivery robots.

H1B Sponsorship

Serve Robotics has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2023 (4)
2022 (1)
2021 (5)

Funding

Current Stage
Public Company
Total Funding
$394M
Key Investors
PostmatesNVIDIANeo
2025-10-10Post Ipo Equity· $100M
2025-01-07Post Ipo Equity· $80M
2024-12-01Post Ipo Equity· $86M

Leadership Team

leader-logo
Ali Kashani
Co-founder & CEO
linkedin
leader-logo
Dmitry Demeshchuk
Co-Founder and VP of Software
linkedin
Company data provided by crunchbase