Staff Machine Learning Engineer
$190,000–$285,000 year
On-site · San Francisco, California, United States
San Francisco, California, United StatesOn-siteFull Time$190,000–$285,000 yearMid LevelNot SpecifiedData ServicesEnterprise
Type
Full Time
Level
Mid Level
Education
Not Specified
Company size
Enterprise
Industry
Data Services
Job Summary
Drive the development and deployment of AI models and systems at Databricks. Responsibilities include design and implementation of ML pipelines, collaboration with AI researchers and product teams, and building scalable backend systems for GenAI products. Requires 2-8 years of experience, expertise in language modeling technologies, proficiency in Python and ML frameworks like TensorFlow/PyTorch, along with strong analytical skills and coding abilities.
Required Qualifications
- 2-8 years of machine learning engineering experience
- Strong track record of working with language modeling technologies
- Proficiency in Python, TensorFlow/PyTorch, and scalable ML architectures
- Ability to drive end-to-end model development
- Strong analytical and problem-solving skills
- Strong coding and software engineering skills
Desired Qualifications
- Experience with LLM fine-tuning
- Familiarity with prompt engineering
- Knowledge of retrieval-augmented generation (RAG)
Additional Requirements
- Individuals are considered without regard to protected characteristics
Apply with one swipe on Sorce. We auto-fill applications and apply on your behalf — no cover letters, no 40-minute forms.
Hiring someone like this?
Get your role in front of qualified candidates on Sorce.