Build end-to-end ML software for autonomy foundation models: data pipelines, training/evaluation workflows, model implementation (LLM/VLM/VLA), simulator integration, and low-latency serving and inference tooling for vehicle deployment.
About us
We’re building the next generation of ground transportation with advanced physical AI to simplify the toughest challenges in modern freight. Our stealth team, founded by the engineers who scaled autonomous driving, is developing an entirely new vehicle platform. We move fast, stay tightly aligned between engineering and product, and focus on creating reliable, real-world autonomous systems.
What You’ll Do
- Build the software backbone for autonomy-focused foundation models: design and ship multimodal data pipelines (ingest, validate, shard, package) and reproducible training/evaluation workflows (manifests, checkpoints, failure handling).
- Implement and iterate on LLM, VLM, and VLA architectures; own model code paths, input/tokenization, inference runners, and output heads for downstream consumers.
- Integrate and operate simulators for closed-loop evaluation; build tooling for metrics, visualization, and experiment management.
- Deliver production-grade serving and inference tooling for deterministic, low-latency operation on bench/mule and eventual vehicle deployments.
- Own systems from scratch: architecture → implementation → testing → documentation → iteration; raise the bar on code quality, reliability, and observability.
What We’re Looking For
- Education & Experience: MS in CS/ML/Robotics or BS + ≥2 years building ML/data/evaluation systems
- Software engineering excellence: Strong Python fundamentals (data structures, testing, debugging, modular design) and a track record of shipping production-quality code/APIs and reliable automation.
- Pipelines → Training → Serving: Demonstrated experience building data/ML pipelines and evaluation tooling, and integrating training and inference using PyTorch, TensorFlow, or JAX.
- Datasets at scale: Dataset packaging, sharding, manifest formats, and integrity checks for large multimodal datasets.
- Performance & optimization: Practical work improving training/inference throughput and latency (e.g., mixed precision, efficient batching, model parallelism).
- MLOps & infrastructure: Cloud storage and training workflows, containerization, CI/CD, and experiment observability (tracking, logging, metrics).
- Team fit: Strong communication, collaborative with research and engineering partners, and a bias for ownership/independence in a small, fast-moving team.
- Nice to have: Prior work on perception, detection, or multimodal models.
Similar Jobs
Fintech • Healthtech • Analytics
Own the production runtime for Phare’s machine learning stack by deploying, serving, and scaling models across GPU inference endpoints and batch or streaming workflows. Build progressive delivery pipelines, automated rollbacks, observability, alerting, and incident response processes. Manage model registries, retraining lifecycles, versioning, data lineage, security controls, and reproducible releases using Kubernetes, Terraform, Docker, and CI/CD.
Top Skills:
APIsCi/CdData LineageDockerEncryptionGpusKubernetesLogsMetricsModel RegistriesModel TelemetryObservabilityRbacSecrets ManagementTerraformTraces
Social Media
Set technical vision for Pinterest’s Product ML Infrastructure, leading distributed model training, fine-tuning, evaluation, and high-scale CPU/GPU inference. Improve GPU efficiency, data loading, execution, kernels, memory, compilation, quantization, scheduling, and capacity. Build reliable, observable training and serving platforms, drive architecture decisions and migrations, partner with AI/ML teams, mentor senior engineers, and establish operational standards across the organization.
Top Skills:
Ads RankingC++CompilationDistributed Machine Learning SystemsFeature PlatformsGpuGpu KernelsJavaModel TrainingOnline InferencePythonQuantizationRecommender SystemsRetrieval Systems
Artificial Intelligence • Consumer Web • Digital Media • Machine Learning • Software
Build and maintain Quora’s machine learning platform and distributed model-serving infrastructure. Develop scalable systems for recommendation and language models, optimize GPU serving latency, throughput, and cost, improve ML tooling and feature stores, and contribute to platform modernization. The role includes production deployment, collaboration with senior engineers, mentoring, and participation in an on-call rotation.
Top Skills:
Amazon EksAWSC++Distributed SystemsDockerGoGpu ServingKubernetesNvidia TritonPythonPyTorchRayTensorFlow
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
