Databricks Logo

Databricks

Staff Backend Software Engineer- (AI Platform)

Posted 16 Days Ago
Be an Early Applicant
In-Office
San Francisco, CA
166K-225K Annually
Senior level
In-Office
San Francisco, CA
166K-225K Annually
Senior level
As a Staff Backend Software Engineer on the AI Platform team, you will build infrastructure for AI applications, improve distributed AI workloads, and collaborate with various teams to enhance developer and data scientist experiences.
The summary above was generated by AI

P-1428

At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building and running the world's best data and AI infrastructure platform so our customers can use deep data insights to improve their business. Founded by engineers — and customer obsessed — we leap at every opportunity to solve technical challenges, from designing next-gen UI/UX for interfacing with data to scaling our services and infrastructure across millions of virtual machines. And we're only getting started.

As part of the AI Platform team, you’ll build the substrate that powers everything from data apps, AI agents, model training, model serving, and vector search.  You’ll be joining a high-agency, high-visibility team operating at the frontier of AI infrastructure — with deep ties to research, product, and real-world enterprise use cases.  Databricks Mosaic AI is one of our fastest-growing businesses helping thousands of our customers democratize AI within their organizations. We’re building the infrastructure that powers the next generation of AI. 

We’re hiring across multiple teams in our AI Engineering org including:  

The impact you will have:
  • Build infrastructure that powers our flagship offerings like  MLflow, AI Gateway, Databricks Apps, Agent Framework, Agent Bricks, and Foundation Model APIs, to state a few.
  • Improve reliability, latency, and efficiency of distributed AI workloads
  • Collaborate with platform, infra, and ML teams to deliver seamless end-to-end experiences
  • 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
  • Strong programming skills in Scala, Go, or Python
  • 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
Bonus points for:
  • Experience with real-time serving, ML infrastructure, or GPU orchestration
  • Exposure to platforms like SageMaker, Vertex AI, or Azure ML
  • Contributions to OSS projects like MLflow, PyTorch, or Ray
  • Built developer platforms or internal tools supporting AI workflows

Pay Range Transparency

Databricks is committed to fair and equitable compensation practices. The pay range(s) for this role is listed below and represents the expected salary range for non-commissionable roles or on-target earnings for commissionable roles.  Actual compensation packages are based on several factors that are unique to each candidate, including but not limited to job-related skills, depth of experience, relevant certifications and training, and specific work location. Based on the factors above, Databricks anticipates utilizing the full width of the range. The total compensation package for this position may also include eligibility for annual performance bonus, equity, and the benefits listed above. For more information regarding which range your location is in visit our page here.


Local Pay Range
$166,000$225,000 USD

About Databricks

Databricks is the data and AI company. More than 10,000 organizations worldwide — including Comcast, Condé Nast, Grammarly, and over 50% of the Fortune 500 — rely on the Databricks Data Intelligence Platform to unify and democratize data, analytics and AI. Databricks is headquartered in San Francisco, with offices around the globe and was founded by the original creators of Lakehouse, Apache Spark™, Delta Lake and MLflow. To learn more, follow Databricks on Twitter, LinkedIn and Facebook.
Benefits
At Databricks, we strive to provide comprehensive benefits and perks that meet the needs of all of our employees. For specific details on the benefits offered in your region, please visit https://www.mybenefitsnow.com/databricks. 

Our Commitment to Diversity and Inclusion

At Databricks, we are committed to fostering a diverse and inclusive culture where everyone can excel. We take great care to ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Individuals looking for employment at Databricks are considered without regard to age, color, disability, ethnicity, family or marital status, gender identity or expression, language, national origin, physical and mental ability, political affiliation, race, religion, sexual orientation, socio-economic status, veteran status, and other protected characteristics.

Compliance

If access to export-controlled technology or source code is required for performance of job duties, it is within Employer's discretion whether to apply for a U.S. government license for such positions, and Employer may decline to proceed with an applicant on this basis alone.

Top Skills

Agent Bricks
Agent Framework
Ai Gateway
Databricks Apps
Foundation Model Apis
Go
Mlflow
Python
Scala

Similar Jobs

16 Days Ago
In-Office
192K-260K Annually
Expert/Leader
192K-260K Annually
Expert/Leader
Big Data • Machine Learning • Software • Analytics • Big Data Analytics
The Staff Backend Software Engineer will design and implement systems for Databricks Model Serving, focusing on scalability, reliability, and operational excellence in serving AI/ML models.
Top Skills: AutoscalingCpuGpuInference SystemsLarge-Scale Distributed SystemsModel ServingObservabilityRoutingScheduling
16 Days Ago
In-Office
192K-260K Annually
Expert/Leader
192K-260K Annually
Expert/Leader
Big Data • Machine Learning • Software • Analytics • Big Data Analytics
As a Staff Backend Software Engineer, you will design and implement systems for high-scalability and low-latency inference using AI models, driving architectural decisions and mentoring teams.
Top Skills: APIsDeep LearningGpu Workloads
16 Days Ago
In-Office
166K-225K Annually
Senior level
166K-225K Annually
Senior level
Big Data • Machine Learning • Software • Analytics • Big Data Analytics
As a Senior Engineer at Databricks, you'll design scalable systems for AI/ML model serving, improve performance, and lead technical initiatives.
Top Skills: AIAPIsCpuDistributed SystemsGpuMl

What you need to know about the Boston Tech Scene

Boston is a powerhouse for technology innovation thanks to world-class research universities like MIT and Harvard and a robust pipeline of venture capital investment. Host to the first telephone call and one of the first general-purpose computers ever put into use, Boston is now a hub for biotechnology, robotics and artificial intelligence — though it’s also home to several B2B software giants. So it’s no surprise that the city consistently ranks among the greatest startup ecosystems in the world.

Key Facts About Boston Tech

  • Number of Tech Workers: 269,000; 9.4% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Thermo Fisher Scientific, Toast, Klaviyo, HubSpot, DraftKings
  • Key Industries: Artificial intelligence, biotechnology, robotics, software, aerospace
  • Funding Landscape: $15.7 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Summit Partners, Volition Capital, Bain Capital Ventures, MassVentures, Highland Capital Partners
  • Research Centers and Universities: MIT, Harvard University, Boston College, Tufts University, Boston University, Northeastern University, Smithsonian Astrophysical Observatory, National Bureau of Economic Research, Broad Institute, Lowell Center for Space Science & Technology, National Emerging Infectious Diseases Laboratories

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account