Senior ML & AI Technical Solutions Engineer

Cacheflow; Bengaluru; On-site; INR 2,000,000 - 3,000,000; Full time

Benefits offered by this job

Comprehensive benefits

Diversity and inclusion commitment

Job summary

Cacheflow is looking for a Senior ML & AI Technical Solutions Engineer based in Bengaluru, India. This role focuses on helping customers debug and optimize their ML workloads on the Databricks platform, providing technical expertise, and ensuring top-notch customer service.

The ideal candidate will have over 8 years of experience with ML and AI systems and proficiency in multiple programming languages including Python and Scala. A commitment to diversity and inclusion is central to Cacheflow’s culture.

Qualifications

• 8+ years of experience designing, building, and scaling Data, ML, and AI systems in production.
• Expertise in Machine Learning and/or Generative AI.
• Prior support or customer-facing experience is advantageous.

Responsibilities

• Act as senior technical solution expert for complex issues in ML and AI applications.
• Analyze and troubleshoot production workloads at the code level.
• Guide customers on leveraging Databricks AI for Generative AI use cases.

Skills

Experience in designing ML/AI systems

Proficiency in Python, Scala, and Java

Knowledge of MLOps and LLMOps

Expertise in feature engineering

Experience with AWS, Azure, or GCP

Education

Bachelor’s or Master’s degree in Computer Science or related field

Tools

Apache Spark

Databricks platform

Job description

Senior ML & AI Technical Solutions Engineer

Mission

As a Senior ML & 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. TSE’s have proven production troubleshooting and optimization 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 output of the team. 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.
• Analyze and troubleshoot production workloads at the code level, optimize 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, labeling, 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 to teach 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 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, optimization, 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 a must for this role, but the ability and desire to develop excellent customer service skills is.
• Prior experience as 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 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 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.