Join Our Team as a Sr. Machine Learning Engineer - GenAI & LLM at Databricks

Databricks, a leader in data and AI technology, is seeking a dedicated Sr. Machine Learning Engineer with a specialization in General AI and Large Language Models (LLM). This role offers the flexibility to work remotely, with a preference for candidates located in the Bay Area, although all qualified candidates across the U.S. will be considered.

About the Role

The Machine Learning (ML) Practice Team at Databricks is looking for a professional to join our dynamic group. As part of our customer-facing team, you will tackle the increasing demand for solutions based on Large Language Models (LLM), allowing you to work on cutting-edge technologies in LLMs, MLOps, and ML.

In this role, you will:

• Develop LLM solutions tailored to customer data, enhancing enterprise knowledge repositories and facilitating natural language interactions with structured data.
• Enhance and optimize data science workloads for our customers, ensuring the effective application of MLOps to bring these workloads to live environments across various domains.
• Offer guidance on data science best practices in areas such as architecture and tool implementation.
• Engage in knowledge sharing by presenting at top conferences like the Data+AI Summit and providing mentorship within the ML Subject Matter Expert community at Databricks.
• Work closely with product and engineering teams to steer product priorities and development, directly influencing the future roadmap.

Who You Are

We are looking for candidates who possess:

• Hands-on experience with modern natural language processing techniques, including the deployment of LLMs using tools like HuggingFace, Langchain, and OpenAI.
• A minimum of 4 years’ experience in data science, with proficiency in ML tools like pandas, scikit-learn, and TensorFlow/PyTorch.
• Experience in deploying machine learning solutions on major cloud platforms like AWS, Azure, or GCP.
• An advanced degree in a quantitative field such as Computer Science or Engineering, or the equivalent practical experience.
• Excellent communication skills, capable of conveying complex technical concepts to diverse audiences.
• A passion for collaborative problem-solving, continuous learning, and delivering value through innovative technology.
• [Preferred] Experience with Apache Spark™ for managing large-scale data sets.

Benefits at Databricks

As part of our commitment to employee well-being, we offer:

• Comprehensive health benefits including Medical, Dental, and Vision plans.
• Retirement plans with 401(k) options.
• Generous time-off policies and flexible schedules.
• Fitness reimbursements and home office setup support.
• Professional development funds and more.

Pay Range Transparency

Our commitment to fair and equitable compensation means the salary range for this role is $124,800—$220,800 USD, varying by location and candidate experience. This position may also be eligible for other compensation, such as bonuses and equity awards.

About Databricks

Databricks assists over 10,000 organizations globally in unifying and democratizing their data, analytics, and AI. Learn more about us and our mission to innovate by following us on social platforms like , , and .

Commitment to Diversity and Inclusion

Databricks is devoted to fostering an inclusive environment where diverse talent thrives. Our hiring practices are designed to be inclusive and equitable across all spectrums.

Compliance Notice

This role may require access to export-controlled technology, and any offer made could be contingent upon our ability to secure the necessary licenses.

Ready to shape the future of AI? Apply now to become a Sr. Machine Learning Engineer at Databricks and help turn data into breakthroughs.