Staff Machine Learning Engineer
San Francisco, California, United States · Posted 6 months agoDescription
P-1504
The Applied AI team at Databricks sits at the forefront of advancing GenAI-powered products. Over the past years, we’ve launched Databricks Assistant, AI/BI Genie, and Agent Bricks 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 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
- 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).
- Develop novel data collection, fine-tuning, and LLM technologies that achieve optimal performance on specific tasks and domains.
- Design and implement ML pipelines for data preprocessing, feature engineering, model training, hyperparameter tuning, and model evaluation, enabling rapid experimentation and iteration.
- 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.
- Build scalable, reusable backend systems to support GenAI products across the company. Develop robust logging, telemetry, and evaluation harnesses to ensure reliable model performance.
What We’re Looking For
- 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.
- 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.
- Proficiency in Python, TensorFlow/PyTorch, and scalable ML architectures.
- Ability to drive end-to-end model development, from research and prototyping to deployment and monitoring.
- Strong analytical and problem-solving skills, with a passion for improving AI-driven user experiences.
- Strong coding and software engineering skills, and familiarity with software engineering principles around testing, code reviews and deployment.
- Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG) is a bonus.
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!
Please note we are open to employees working from our Mountain View, CA office for this position.
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. 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 here.
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.
Similar jobs
Est. 222,325 USD
P-1607At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We…
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 w…
Est. 195,500 USD
P-160 Who We Are Our GenAI observability and quality product provides advanced monitoring and insights for GenAI systems, giving customers real-time visibility into their system's performance, along with a suite of tools…
Est. 225,000 USD
Who we are At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthrough…
Est. 216,104 USD
PLEASE NOTE: Due to federal contract requirements and client site access obligations, U.S. citizenship and eligibility for a U.S. government secret clearance are required to access classified information. The position is…
Est. 222,325 USD
P-1607At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We…
Est. 211,850 USD
P-1285 About This Role As a staff software engineer for GenAI inference, you will lead the architecture, development, and optimization of the inference engine that powers Databricks Foundation Model API.. You’ll bridge r…
Est. 225,625 USD
P-1592 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthro…
Est. 174,125 USD
SLSQ327R637 At Databricks, we are passionate about enabling GTM Analytics Engineering builds the internal data and AI platform that Databricks' own go-to-market organization runs on — used by 6,000+ field users across sa…
P-1439 As a Senior Applied ML/AI Engineer at Databricks, you will apply machine learning and optimization algorithms to improve the usability and efficiency of the current AutoML and several other user-facing products th…
Est. 192,488 USD
P-1591 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthro…
Est. 226,000 USD
RDQ427R109 At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthrough…
Est. 193,000 USD
P-78 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroug…
Est. 195,500 USD
P-1428 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthro…
Est. 227,500 USD
P-1930 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthro…
Est. 181,528 USD
AI Engineer - FDE (Forward Deployed Engineer) (ALL LEVELS) CSQ327R177 Mission The AI Forward Deployed Engineering (AI FDE) team is a highly specialized customer-facing AI team at Databricks. We deliver professional servi…
Est. 188,125 USD
RDQ127R59 Summary As a Senior Applied ML Engineer on the Applied AI team at Databricks, you will use machine learning, scheduling, and optimization algorithms to maximize the efficiency and performance of our infrastruct…
Est. 227,500 USD
P-1930 At Databricks, we are passionate about enabling data and AI teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthro…
Est. 177,175 USD
RDQ427R542 At Databricks, we are passionate about enabling data teams to solve the world's most challenging problems — from making the next mode of transportation a reality to accelerating medical breakthroughs. We do th…
Est. 159,550 USD
RDQ427R542 At Databricks, we are passionate about enabling data teams to solve the world's most challenging problems — from making the next mode of transportation a reality to accelerating medical breakthroughs. We do th…
Est. 226,000 USD
P-57 At Databricks, we are obsessed with enabling data teams to solve the world's toughest problems. We do this by building and running the world's best data and AI infrastructure platform, so our customers can focus on…
P-1502 At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. W…
Est. 192,500 USD
P-1428 At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. W…
Est. 195,500 USD
At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do th…
Est. 213,000 USD
P-1381 At Databricks, we are passionate about enabling Data & AI teams to solve the world's toughest problems - from making the next mode of transportation a reality to accelerating the development of medical breakth…
Est. 305,000 USD
Principal AI Research Scientist, Research Director - AI Scaling P-1227 About Databricks AI At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems, from security threat detection to…
Est. 170,600 USD
At Databricks, we are obsessed with enabling data teams to solve the world's toughest problems, from security threat detection to cancer drug development. We do this by building and running the world's best data and AI i…
Est. 226,000 USD
At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do th…
Est. 213,750 USD
FEQ427R382 As an AI/ML Specialist Solutions Architect (SSA), you will lead the advanced AI/ML technical strategy for your customers — owning complex architecture discussions, driving platform adoption, and serving as a t…
Est. 120,000 GBP
ReqID: FEQ327R501 Location: London Skills: Data Science, Machine Learning, AI, LLM, GenAI Mission As a Senior Specialist Solutions Architect (ML & AI), you will serve as the trusted technical ML and AI expert for Dat…