Staff Software Engineer- Databricks AI

Neon; Bengaluru; On-site; INR 4,000,000 - 7,000,000; Full time

Job summary

Databricks in Bengaluru is expanding its AI engineering team and is seeking a Staff Software Engineer to lead the new site. You will set platform strategy and architecture, drive design reviews, and mentor senior engineers while staying hands-on to deliver a world-class MLOps platform.

You will align product, research, and engineering across teams, shape APIs and developer experiences, and help customers ship AI safely and at scale.

Qualifications

• BS or higher in Computer Science or a related field.
• 10+ years in distributed systems or platform development.
• Strong background in algorithms and data structures.
• Proven technical leadership and end-to-end delivery experience.

Responsibilities

• Define platform strategy and long-term architecture.
• Lead engineering from design to production rollout with high performance and reliability.
• Solve hard problems in ingesting traces, metrics, and artifacts at scale.
• Align product, research, and engineering across sites on direction and trade-offs.
• Mentor engineers through design reviews and technical discussions.

Skills

Scala

Java

Distributed systems

System design

Cloud technologies

Security

Operational excellence

Mentoring

Communication

ML/LLM basics

Education

BS in Computer Science

Tools

Docker

Kubernetes

AWS

Azure

GCP

Job description

At Databricks, we are passionate about helping enterprises that build ML models and AI agents deliver high-quality AI to production fast. Founded by engineers, and customer obsessed, we leap at every opportunity to tackle technical challenges, from designing great developer experiences to building reliable systems at global scale. And we're only getting started.

Our team builds the MLOps and LLMOps platform that millions of AI developers and ML practitioners use to track model training experiments, manage model registry, observe and evaluate AI agents. We're the team behind MLflow, the most widely adopted open-source AI engineering platform with more than 30 million package downloads per month, and its managed service used by thousands of organizations globally. We're now expanding to Bengaluru, India, and building our first AI engineering team there from scratch to work on this platform.

As a Staff Software Engineer and a founding member of the Bengaluru team, you'll be the team's technical lead. You'll set the platform strategy and architecture, drive alignment with partner teams, and lead the team to deliver a world-class MLOps and LLMOps platform, while staying hands-on.

Check out our recent blog post: coSTAR: How we ship AI agents at Databricks fast, without breaking things

The impact you will have:

• Define the platform strategy and long-term architecture.
• Lead engineering efforts from design to production rollout, with high standards for performance, correctness, and reliability.
• Solve hard problems in ingesting and querying traces, metrics, and artifacts at scale, and in evaluating AI agents in production.
• Align product, research, and engineering teams across sites on technical direction and trade-offs.
• Shape APIs and developer experiences that make it easy for customers to ship AI.
• Mentor senior engineers through design reviews and technical discussions.
• Represent the team externally through blog posts, talks, and open-source contributions to popular AI frameworks.

What we look for:

• BS (or higher) in Computer Science, or a related field.
• 10+ years of experience developing large-scale distributed systems, scaling infrastructure, or building platforms and developer tools for other teams.
• Strong foundation in algorithms, data structures, and system design.
• Strong experience with backend technical stacks such as Scala or Java.
• Experience with cloud technologies, e.g. AWS, Azure, GCP, Docker, Kubernetes.
• Engineering maturity in security, correctness, and operational excellence.
• Proven track record as a technical lead, driving architecture decisions and delivering complex, high-impact projects end-to-end.
• Ability to lead through influence across teams.
• Strong communication skills and a passion for mentoring engineers.
• Experience with ML or LLM systems is a plus.

About Databricks

Databricks is the Data and AI company. More than 20,000 organizations worldwide — including adidas, AT&T, Bayer, Block, Mastercard, Rivian, Unilever, and 70% of the Fortune 500 — rely on the Databricks Data + AI Platform to build and scale data and AI apps, analytics and agents. Headquartered in San Francisco with 30+ offices around the globe, Databricks offers a unified platform that includes Genie, Lakebase, Agent Bricks, Lakeflow, Lakehouse, and Unity Catalog. To learn more, follow Databricks on LinkedIn, X, YouTube, and Instagram.

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.

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.