GenAI Staff Machine Learning Engineer, AI Runtime

Databricks, a leading data and AI company serving over 10,000 organizations worldwide, is seeking a Staff Machine Learning Engineer for their AI Runtime team. This role, based in San Francisco, focuses on developing cutting-edge tooling and technologies for foundation models. The position emerged following Databricks' acquisition of MosaicML in July 2023, combining expertise in large-scale AI model training and deployment.

The role involves designing and implementing state-of-the-art tools for foundation model development, solving complex challenges in data preprocessing and model training, and optimizing resource utilization while maintaining performance. You'll work with advanced technologies like PyTorch and various GPU platforms, contributing to significant open-source projects such as Composer, Streaming, and LLM Foundry.

As a Staff ML Engineer, you'll collaborate across teams, contribute to the AI community through research and publications, and help shape the future of AI infrastructure. The ideal candidate brings deep expertise in machine learning frameworks, distributed training systems, and a proven track record of building scalable AI solutions.

The position offers competitive compensation ($192,000-$260,000 base salary), equity, and comprehensive benefits. This is an exceptional opportunity to work at the forefront of AI technology, contributing to projects that enable organizations worldwide to leverage advanced AI capabilities. The role combines technical depth with strategic impact, making it ideal for experienced engineers passionate about advancing the field of machine learning.