Senior ML & AI Technical Solutions Engineer

Reposted One Month Ago
Be an Early Applicant
Bengaluru, Bengaluru Urban, Karnataka, IND
In-Office
Senior level
Big Data • Machine Learning • Software • Analytics • Big Data Analytics
The Role
Provide senior-level technical expertise to debug, optimise, and productionize GenAI and ML workloads on Databricks. Troubleshoot data and ML pipelines, LLM deployments, model serving, vector search, agent systems, monitoring, and cost/latency issues. Advise customers on MLOps/LLMOps, experiment tracking, model lifecycle, prompt engineering, retrieval/embedding systems, and collaborate with engineering and account teams to influence product and create documentation.
Summary Generated by Built In

P-1377

Mission

As a Senior ML and AI Technical Solutions Engineer, you play a critical role by helping customers debug and maintain stable GenAI and ML Workloads with AI agent systems using the Databricks Platform.  You will develop product expertise end-to-end by advising a broad set of customers and use cases across the space - including products such as Agent Bricks, Vector Search and Model Serving.  You will collaborate cross-functionally with other teams - whether that’s working with engineering to improve the product or interacting directly with the account team on a specific customer issue.  TSEs have proven production troubleshooting and optimisation experience to help our customers’ workloads run smoothly and to achieve their strategic objectives with ML/AI technology with Databricks.  Additionally, you are an early adopter of GenAI technology to improve your own efficiency and amplify the team's output.  Reporting to a TSE manager - you will be part of a world class global support engineering organization for Databricks, known for your technical depth and delivering impeccable customer service.

The Impact You Will Have
  • Act as senior technical solution expert for complex issues spanning data pipelines, ML pipelines and/or AI applications, applying deep expertise in distributed systems. 
  • Analyse and troubleshoot production workloads at the code level, optimise for performance, reliability, latency, and cost.
  • Diagnose and support Machine Learning and/or Large Language Model deployments, including real-time and batch inference, autoscaling, monitoring, logging, and alerting. Serve as a Subject Matter Expert guiding customers on experiment tracking, model registry, versioning, evaluation, labelling, tracing, and lifecycle observability.
  • Provide high-quality support by guiding customers in leveraging Databricks AI to solve generative AI use cases & challenges, leveraging LLMs, MCP, AI Agents, RAG/Agentic RAG, APIs, vector embeddings, semantic search, Vector Search/Lakebase databases, context orchestration, memory management, and prompt engineering.
  • Collaborate with internal teams to influence roadmap, product improvements and support business growth.
  • Develop expertise in productionizing systems in Databricks and share your knowledge by contributing to wikis and other technical documentation, or by teaching our AI systems new skills, which will be used internally and externally by customers and partners.

What We Look For

8+ years of experience designing, building, and scaling Data, Machine Learning, and AI systems on-premises and in the cloud using Python, Scala, and Java in production environments, with expertise in Machine Learning and/or generative AI. Experience with cloud platforms (AWS, Azure, or GCP); familiarity with Databricks is a plus. Proficient in data engineering necessary for orchestrating end-to-end machine learning training pipelines, ideally with experience processing large datasets with Apache Spark.

  • SME knowledge in feature engineering, ML frameworks, model training, model monitoring, drift detection, and retraining strategies. Proficient in working with algorithms and deep learning, along with NLP techniques.
  • Prior experience building, designing or troubleshooting LLM-based Generative AI applications. Familiarity with agentic frameworks (e.g., LangChain, LangGraph etc). Expertise in context orchestration, including prompt design, memory management, retrieval systems, vector embeddings, semantic search, and tool integrations.
  • Comprehensive Knowledge of MLOps and LLMOps with expertise in model evaluation, scoring, ranking, optimisation, training, validation, and packaging.
  • Experience developing agent skills, plugins, and debugging with native AI capabilities is a plus. 
  • Prior support or customer-facing experience is not required for this role, but the ability and desire to develop excellent customer service skills are. 
  • Prior experience in Data Scientist, ML Engineer, or AI Engineer roles is highly valued.
  • Bachelor’s or Master’s degree in Computer Science, Engineering, or related field (or equivalent experience). Professional certifications are good to have.

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 click here.

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.

Skills Required

  • 8+ years designing, building, and scaling data, ML, and AI systems in production
  • Production experience with Python
  • Production experience with Scala
  • Production experience with Java
  • Experience with cloud platforms (AWS, Azure, or GCP)
  • Proficient data engineering skills and experience processing large datasets with Apache Spark
  • SME knowledge in feature engineering, model training, monitoring, drift detection, and retraining strategies
  • Experience building or troubleshooting LLM-based generative AI applications and NLP techniques
  • Familiarity with agentic frameworks and context orchestration (e.g., LangChain, LangGraph), vector embeddings, semantic search, and retrieval systems
  • Comprehensive knowledge of MLOps and LLMOps including model evaluation, packaging, deployment, and observability
  • Familiarity with Databricks
  • Experience developing agent skills, plugins, or debugging native AI capabilities
  • Bachelor's or Master's degree in Computer Science, Engineering, or related field (or equivalent experience)
  • Professional certifications in relevant areas

Databricks Compensation & Benefits Highlights

The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Databricks and has not been reviewed or approved by Databricks.

  • Healthcare Strength — Company materials highlight comprehensive medical, dental, and vision coverage alongside mental-health resources, wellness reimbursements, and business travel insurance. Offerings are described as broad and modern, with core health coverage consistently emphasized.
  • Parental & Family Support — Paid parental leave is explicitly called out, with details such as up to 20 weeks for birthing parents and up to 12 weeks for non-birthing parents in the U.S. Public materials also reference family-forming support, reinforcing the focus on families.
  • Wellbeing & Lifestyle Benefits — Wellness programs and perks include gym reimbursement, periodic wellness events (e.g., yoga, massages), and in-office meals and snacks in many locations. Personal development funds and discounts further enhance lifestyle and growth support.

Databricks Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
2,200 Employees
Year Founded: 2013

What We Do

As the leader in Unified Data Analytics, Databricks helps organizations make all their data ready for analytics, empower data science and data-driven decisions across the organization, and rapidly adopt machine learning to outpace the competition. By providing data teams with the ability to process massive amounts of data in the Cloud and power AI with that data, Databricks helps organizations innovate faster and tackle challenges like treating chronic disease through faster drug discovery, improving energy efficiency, and protecting financial markets.

Similar Jobs

Ericsson Logo Ericsson

Master 5G baseband Developer

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Bangalore, Bengaluru Urban, Karnataka, IND
88000 Employees

Ericsson Logo Ericsson

SI

Cloud • Information Technology • Internet of Things • Machine Learning • Software • Cybersecurity • Infrastructure as a Service (IaaS)
In-Office
Bangalore, Bengaluru Urban, Karnataka, IND
88000 Employees

Boeing Logo Boeing

Software Engineer

Aerospace • Information Technology • Software • Cybersecurity • Design • Defense • Manufacturing
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
170000 Employees

Atlassian Logo Atlassian

Solutions Engineer

Cloud • Information Technology • Productivity • Security • Software • App development • Automation
In-Office or Remote
Bengaluru, Bengaluru Urban, Karnataka, IND
11000 Employees

Similar Companies Hiring

Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees
Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
60 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account