Most Loved Workplace® Certified Job

Staff Machine Learning Engineer

Most Loved Workplace® Certified Job

92% of candidates apply because they are a Most Loved Workplace®

San Francisco, California

Engineering - Pipeline

About the Role

At Databricks, a Most Loved Workplace® certified employer in the Technology space, Democratizing data and AI for every organization worldwide.

P-1504

The Applied AI team at Databricks sits at the forefront of advancing GenAI-powered products. Over the past years, we’ve launched&nbsp;<a class="c-link" href="https://www.Databricks.com/blog/introducing-Databricks-assistant" target="_blank" data-stringify-link="https://www.Databricks.com/blog/introducing-Databricks-assistant" data-sk="tooltip_parent">Databricks Assistant</a>,&nbsp;<a class="c-link" href="https://www.Databricks.com/product/ai-bi/genie" target="_blank" data-stringify-link="https://www.Databricks.com/product/ai-bi/genie" data-sk="tooltip_parent">AI/BI Genie</a>, and&nbsp;<a class="c-link" href="https://www.Databricks.com/blog/introducing-agent-bricks" target="_blank" data-stringify-link="https://www.Databricks.com/blog/introducing-agent-bricks" data-sk="tooltip_parent">Agent Bricks&nbsp;</a>working with product teams, and made significant strides in LLM quality for these products. These products are used by 100s of thousands of Databricks users every day. We are tackling challenging problems like code suggestion, error detection and correction, text-to-sql generation, automatic pipeline generation, knowledge QA and many others.

As our GenAI products continue to evolve, we are seeking multiple &nbsp;GenAI Engineers from junior levels to more senior levels to drive the next phase of development. In 2025, we will focus on enhancing LLM quality, expanding GenAI capabilities across Databricks products, and strengthening our platform architecture to enable seamless AI interactions at scale.

Key Responsibilities

<ul>

<li>Shape the direction of our applied AI areas and intelligence features in our products. Drive the development and deployment of state-of-the-art AI models and systems that directly impact the capabilities and performance of Databricks' products and services (e.g., Databricks Assistant and AI/BI Genie).</li>

<li>Develop novel data collection, fine-tuning, and LLM technologies that achieve optimal performance on specific tasks and domains.</li>

<li>Design and implement ML pipelines for data preprocessing, feature engineering, model training, hyperparameter tuning, and model evaluation, enabling rapid experimentation and iteration.</li>

<li>Work closely with cross-functional teams, including AI researchers, ML engineers, and product teams, to deliver impactful AI solutions that enhance user productivity and satisfaction.</li>

<li>Build scalable, reusable backend systems to support GenAI products across Databricks. Develop robust logging, telemetry, and evaluation harnesses to ensure reliable model performance.</li>

</ul>

What We’re Looking For

<ul>

<li>2-8 years of machine learning engineering experience in high-velocity, high-growth companies. Alternatively, a strong background in relevant ML research in academia will be considered as an equivalent qualification.</li>

<li>Strong track record of working with language modeling technologies. This could include the following: Developing generative and embedding techniques, modern model architectures, fine tuning / pre-training datasets, and evaluation benchmarks.</li>

<li>Proficiency in Python, TensorFlow/PyTorch, and scalable ML architectures.</li>

<li>Ability to drive end-to-end model development, from research and prototyping to deployment and monitoring.</li>

<li>Strong analytical and problem-solving skills, with a passion for improving AI-driven user experiences.</li>

<li>Strong coding and software engineering skills, and familiarity with software engineering principles around testing, code reviews and deployment.</li>

<li>Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG) is a bonus.</li>

</ul>

Why Join Us? At Databricks, we are building state-of-the-art AI solutions that redefine how users interact with data and our products. You’ll have the opportunity to shape the future of AI-driven products at Databricks, work with cutting-edge models, and collaborate with a world-class team of AI and ML experts.

If you're excited about pushing the boundaries of AI in real-world applications, we’d love to hear from you!

&nbsp;

Please note we are open to employees working from our Mountain View, CA office for this position.&nbsp;<div class="content-pay-transparency"><div class="pay-input"><div class="description">&nbsp;

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.&nbsp; 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 <a href="https://www.Databricks.com/sites/default/files/2024-08/us-pay-zone-mapping.pdf">here</a>.

&nbsp;</div><div class="title">Local Pay Range</div><div class="pay-range">$190,000&mdash;$285,000 USD</div></div></div><div class="content-conclusion">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&nbsp;<a style="color: rgb(255, 54, 33);" href="https://twitter.com/Databricks" target="_blank" data-saferedirecturl="https://www.google.com/url?q=https://twitter.com/Databricks&amp;source=gmail&amp;ust=1700237575733000&amp;usg=AOvVaw03FL8fJvOD97ytN02f5G2C">Twitter</a>,&nbsp;<a style="color: rgb(255, 54, 33);" href="https://www.linkedin.com/company/Databricks" target="_blank" data-saferedirecturl="https://www.google.com/url?q=https://www.linkedin.com/company/Databricks&amp;source=gmail&amp;ust=1700237575733000&amp;usg=AOvVaw15dLk3q8VxTfHEgCUg7NSt">LinkedIn</a>&nbsp;and&nbsp;<a style="color: rgb(255, 54, 33);" href="https://www.facebook.com/databricksinc" target="_blank" data-saferedirecturl="https://www.google.com/url?q=https://www.facebook.com/databricksinc&amp;source=gmail&amp;ust=1700237575733000&amp;usg=AOvVaw39EcncitnlqV72EG2-RqXJ">Facebook</a>. 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 <a href="https://docs.google.com/document/d/154un3e8Xav4BceOSlcYFZRGEuQI54xMxVydRwQn54eQ/edit?usp=sharing">here</a>.

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.</div>

Want to learn more about what it's like to work at Databricks? View our full profile.

Prefer to apply on the company's career site?

View Original Posting

Why This Is a Most Loved Workplace® Certified Job

Systemic Collaboration

As Databricks continues our rapid growth, we strive to maintain a culture as open and transparent as it was in Databricks’s early days. Making sure employees have regular access to our CEO and co-founders to hear directly about our vision and priorities helps our employees forge a connection between Databricks priorities and their daily work.

Positive Vision for the Future

Databricks’ culture principles are derived from an internal study of the behaviors that have led to high impact work at Databricks. These values have a direct impact on company culture and play a large part in connecting teams across the organization. We believe in being customer-obsessed, truth-seeking, operating from first principles, having a bias for action, and putting Databricks first. Our co-founders are fully involved in day-to-day operations, including business and technical reviews, working alongside our team members to ensure that we are always aligned with our values.

Alignment of Values

Databricks’ CEO and Co-founder, Ali Ghodsi leads by example with a focus on truth-seeking and first principles thinking, two of Databricks’ core values that stay true to Databricks’s origins in academia. He champions the importance of data and reason to question biases and inform decisions. Ali remains hands-on with research and development and hosts company-wide weekly all hands, with a CEO “ask me anytime” session at the beginning of every agenda. His transparency and accessibility are core to the open and collaborative culture of Databricks.

Respect

At Databricks, we cultivate an environment where all employees can bring their unique selves and are empowered to do the best work of their careers. To ensure all voices are heard and ideas are valued, we conduct annual culture and pulse surveys and host all-hands Q&As to collect feedback from all employees. We then report back to Databricks on the specific changes implemented from the feedback collected to ensure employees feel heard and that their feedback is valued. In addition, we invest in our seven Employee Resource Groups (ERGs) that offer support and employee engagement events for individuals from underrepresented backgrounds and allies.

Killer Outcomes

Databricks prioritizes offering an exceptional employee experience by investing in tools, workspaces, and community-building. Our collaboration-first, hybrid model offers flexibility with in-person touchpoints like our annual employee kickoff, weekly CEO all-hands with happy hours, and monthly team building budgets. We are especially focused on maintaining a culture of transparency, honesty, and trust, with leaders actively engaging across channels like Slack and all hands while also championing inclusion efforts through ERG sponsorship. These initiatives help our team feel connected, valued, and empowered to do their best work.

What It's Like to Work Here

zap Bias for Action

lightbulb First Principles Innovation

users Collaborative Excellence

target Truth-Seeking Rigor

compass Generational Mission Focus

Certified for: Global Top 100 Most Loved Workplaces®, Certified Most Loved Workplaces® 2025, Top 100 Most Loved Workplaces® 2025

Frequently Asked Questions About Working at Databricks

Common questions candidates ask about this role and Databricks's workplace

With the Data Intelligence Platform, Databricks democratizes insights to everyone in an organization. Built on an open lakehouse architecture, the Data Intelligence Platform provides a unified foundation for all data and governance, combined with AI models tuned to an organization’s unique characteristics. Now, anyone in an organization can benefit from automation and natural language to discover and use data like experts, and technical teams can easily build and deploy secure data and AI apps and products. With origins in academia and the open source community, Databricks was founded in 2013 by the original creators of the lakehouse architecture and open source projects Apache Spark™, Delta Lake, MLflow and Unity Catalog.

Please review the specific job listing or contact Databricks's recruiting team for details on remote or hybrid work options for this role.

Salary information may vary by role and location. Please check the specific job listing or discuss compensation during the interview process.

Databricks is an established organization in its industry. The company is a certified Most Loved Workplace®, highlighting a strong, positive culture and committed workforce.

Key benefit categories include: Health & Wellness, Work-Life Balance, Office & Lifestyle, Professional Development, Financial. You can view more details on their CertCheck profile.

The day-to-day environment is guided by core values such as Bias for Action and First Principles Innovation and Collaborative Excellence.

The company has committed to inclusive practices including: Customer Obsession with Integrity and Empowering Every Team Member.

They score particularly well in the area of Systemic Collaboration. They have earned Most Loved Workplace® certifications including: Global Top 100 Most Loved Workplaces® and Certified Most Loved Workplaces® 2025.

Databricks currently has 789 open positions. They are hiring across departments like Field Engineering - Other, Delivery Solutions Architects, Enterprise Sales. You can view all current openings at certcheck.mostlovedworkplace.com/companies/databricks/jobs.

The interview process typically involves an initial recruiter screen followed by team interviews. Please contact Databricks's recruiting team for specific details on this role's process.