Software Engineer - E2E Autonomy
Builds ML tools, infrastructure, and manages large datasets for end-to-end autonomy research and productionizing self-driving software. Works with AI research and engineering teams to scale GPU compute, data, and evaluation systems. Requires strong software generalist skills across ML stack.
About the job
Responsibilities
- Build tools and infrastructure to enable large-scale E2E autonomy research led by Dr. Wei Zhan
- Work directly with both AI research and engineering teams to productionize next generation self-driving software
- Help scale end-to-end training by identifying bottlenecks, especially as we continue to scale our GPU compute, onroad data volume, curation efforts, and eval systems
Technologies: Python, Pytorch, CUDA, Bazel, Kubernetes, Spark, Flyte
Requirements
- Exceptional software generalist experience with ambition to contribute up and down the ML stack, onboard the vehicle, and in the cloud
- Strong fundamental skills in software & module design: can tame chaos with libraries & abstractions
- Rapid analytical and problem-solving skills
Nice to Have
- Strong multi-language build system knowledge
- Distributed systems/compute (e.g. k8s, cloud-managed services at scale)
- ML infra fluency - has used and/or contributed to ML and large data systems
Compensation
Base salary range: $153,000 - $222,000 USD annually
Skills
Python, PyTorch, CUDA, Bazel, Kubernetes, Spark, Flyte
Similar jobs
ML Engineering jobs
Applied Intuition
Sunnyvale, CA
Develop and deploy ML-first behavior prediction and planning systems for autonomous vehicles, forecasting the motion and interactions of road users. Requires a bachelor's degree, deep learning lifecycle expertise, and at least three years of production software experience with C++ or Python.
OnePay
United States
Build and operate production AI agents, automation workflows, and integrations that improve complex business processes. The role requires 5+ years of software engineering experience, modern LLM and agent-framework expertise, systems integration skills, and strong cross-functional collaboration.
LangChain
New York, NYBuild and teach reliable AI agent systems through customer workshops, technical content, guidance, and reference implementations. The role requires strong Python and agent-development experience plus a background delivering customer-facing technical training.
Roboflow
San Francisco, CA
Build reinforcement-learning environments, evaluations, datasets, and scalable infrastructure for frontier AI capabilities. The role suits a high-agency generalist engineer with experience in agents, evaluations, or RL workflows and strong communication skills.
Beacon Biosignals
Boston, MADevelop and productionize machine- and deep-learning algorithms for biosignal and EEG data used in medical devices, clinical development, and diagnostics. The role requires 4+ years of industry experience, DSP and statistics expertise, PyTorch proficiency, and familiarity with regulated environments and production ML practices.