Skip to content
Applied IntuitionApplied IntuitionSunnyvale, CA

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.

153k – 222k/yr
On-siteML Engineering

About the role

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

PythonPyTorchCUDABazelKubernetesSparkFlyte

Similar roles

ML Engineering jobs
Writer

AI engineer

WriterSan Francisco, CA +1

Build and deploy scalable AI applications and intelligent agents for enterprise customers. Requires 5+ years of experience with Python, ML frameworks (PyTorch/TensorFlow/JAX), LLMs, and cloud platforms.

152k – 316k/yr
Hybrid5+ YOEML Engineering
Brex

Software Engineer, Forward Deployed Agent Builder

BrexSeattle, WA

Builds and deploys AI agents to automate internal workflows at Brex by embedding with teams, integrating with systems and APIs, and measuring impact. Requires 4+ years experience shipping AI/automation with LLMs, agent frameworks, and databases.

152k – 240k/yr
Hybrid4+ YOEML Engineering
Brex

Software Engineer, Forward Deployed Agent Builder

BrexSan Francisco, CA

Builds and deploys AI agents to automate internal workflows at Brex by embedding with teams, integrating systems, and measuring impact. Requires 4+ years experience with LLMs, agent frameworks, databases, and production AI systems.

152k – 240k/yr
Hybrid4+ YOEML Engineering
Brex

Software Engineer, Forward Deployed Agent Builder

BrexNew York, NY

Builds and deploys AI agents to automate internal workflows at Brex by embedding with teams, integrating systems and APIs, and measuring impact. Requires 4+ years experience shipping AI/automation with LLMs, agent frameworks, and databases.

152k – 240k/yr
Hybrid4+ YOEML Engineering
Notion

AI Applications Engineer

NotionSan Francisco, CA

Develop and prototype AI-driven solutions for GTM, Finance, and People teams, translating business problems into impactful prototypes using ML/AI and LLMs. Requires 4-8 years as Software Engineer or Data Scientist with production AI familiarity.

152k – 207k/yr
Hybrid4+ YOEML Engineering