Databricks · 2 days ago
Sr. Machine Learning Engineer
San Francisco, California
Full-time
Onsite
Mid, Senior Level
$166K/yr - $210K/yr
2+ years expDatabricks is the data and AI company, and they are seeking a Senior Machine Learning Engineer to join their Applied AI team. This role involves driving the development and deployment of state-of-the-art AI models and systems that enhance the capabilities and performance of Databricks' products and services.
AnalyticsArtificial Intelligence (AI)Data StorageInformation TechnologyMachine Learning
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
Qualification
Required
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
Preferred
Experience with LLM fine-tuning, prompt engineering, and retrieval-augmented generation (RAG) is a bonus
Benefits
Eligibility for annual performance bonus
Equity
Company
Databricks
Databricks is a data and AI platform that unifies data engineering, analytics, and machine learning on a lakehouse architecture.
5001-10000 employees
H1B Sponsorship
Databricks has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (385)
2024 (319)
2023 (227)
2022 (222)
2021 (166)
2020 (64)
Funding
Current Stage
Late StageTotal Funding
$25.81BKey Investors
Counterpoint GlobalFranklin TempletonAndreessen Horowitz
2025-12-16Series Unknown· $4B
2025-09-08Series Unknown· $1B
2025-01-13Debt Financing· $5.25B
Leadership Team
Ali Ghodsi
CEO and Co-founder
David Conte
Chief Financial Officer
Recent News
Crunchbase News
2026-01-09
2026-01-09
Company data provided by crunchbase