Senior Machine Learning Engineer (GenAI)

TL;DR

Senior Machine Learning Engineer (GenAI): Building and scaling a customer-facing generative AI platform covering data generation, training, evaluation, serving, and agent-building with an accent on distributed system design, product ownership, and GPU orchestration. Focus on designing robust backend infrastructure, ensuring reliability, security, and scalability of AI platform services.

Location: Remote (US), San Francisco

Salary: $166,000–$225,000 USD

Company

Databricks is a leading data and AI company serving over 10,000 organizations worldwide, including more than half of the Fortune 500, with a platform unifying data, analytics, and AI.

What you will do

• Design and implement end-to-end generative AI platform features for training and serving models
• Collaborate with customers and ML researchers to identify key development areas
• Own product requirements translation into user interfaces and backend distributed systems
• Build core platform infrastructure supporting customer-facing features
• Ensure backend system reliability, security, and scalability

Requirements

• Must have 4+ years programming experience in Python, Scala, Go, or C++
• Strong experience in distributed systems design and large-scale system building
• Experience with ML platform systems for data prep, training, evaluation, and serving
• Product ownership mindset with intuition for system design and usability
• Effective communication skills for cross-disciplinary collaboration

Nice to have

• Direct ML model development experience

Culture & Benefits

• Comprehensive benefits tailored to employee needs
• Commitment to diversity and inclusion
• Global remote work opportunity