Skip to content

Senior Software Engineer (Backend) - AI/ML Environments

166k – 220kMountain View, CAOnsite5+ YOE
Summary

Build AI/ML environment infrastructure enabling researchers to configure training and serving setups reliably. Requires 5+ years backend experience, strong Python/Scala/Java skills, and expertise in distributed systems and containerization.

About the role

The impact you will have

  • Build the infrastructure that enables ML and AI users to configure training and serving environments easily, reliably and reproducibly.
  • Collaborate with other AI infrastructure teams to build the features that customers need to get more from the Databricks platform. Examples include improving performance of setting up virtual environments for short training and data processing sessions, improving observability to help customers debug when runs fail, etc.
  • Interact with turnkey customers and product managers to envision new features and identify areas for improvement.
  • Shape how developers and data scientists build and interact with AI on Databricks.

What we look for

  • 5+ years of experience in backend or infrastructure engineering with a focus on building systems
  • Strong programming skills in Python, Scala or Java
  • Experience with distributed systems, scalable APIs, or cloud-native infrastructure
  • Familiarity with service-oriented architecture, deployment pipelines, and system observability
  • Strong product and ownership mindset — you care about building the right solution, not just any solution
  • Strong understanding of dependency management technologies including virtual environments or containerization technologies.
Skills
PythonScalaJavaDistributed SystemsScalable APIsCloud-Native InfrastructureService-Oriented ArchitectureContainerizationDependency ManagementSystem Observability
Similar roles at this salary range
All Backend Engineering jobs →
Aurelian

Senior Backend Engineer

Senior Backend Engineer to architect and scale voice AI systems for 911 centers, optimizing realtime pipelines and integrating with customer systems. Requires 4+ years of software engineering experience.

150k – 200kSeattle, WABackend EngineeringOn-siteGoSQL
Aurelian

Staff Backend Engineer

Staff Backend Engineer to architect and scale voice AI systems for 911 centers, optimizing realtime pipelines and integrating with customer systems. Requires 6+ years of experience and strong engineering fundamentals.

180k – 240kSeattle, WABackend EngineeringOn-siteGoSQL
ClickUp

Senior Backend Engineer, Platform

Senior backend engineer building and scaling API tooling, shared services, and distributed systems that power ClickUp's AI platform capabilities.

160k – 210kUnited StatesBackend EngineeringRemoteKafkaGolang
Onos Health

Senior Software Engineer

Senior Backend Engineer to architect AI/ML workflows processing healthcare data, lead platform integrations, and own end-to-end feature development. Requires 5+ years experience and prior tech lead experience.

180k – 240kSan Francisco, CABackend EngineeringHybridAWSPython
Tailscale

Software Engineer, Networking

Build and operate Tailscale’s global Funnel and DERP relay infrastructure, focusing on networking features, performance, and reliability. Requires strong Go, networking, and distributed-systems experience.

163k – 226kUnited StatesBackend EngineeringRemoteGoSRE