Senior Machine Learning Engineer - GenAI Platform

Founded in late 2020 by a group of machine learning engineers and researchers, MosaicML allows companies to securely tweak, train, and implement customized AI models on their own data, ensuring maximum security and control. All major cloud providers are compatible with the MosaicML platform, providing maximum flexibility for AI development. Introduced in 2023, MosaicML's pre-trained transformer models have set a new standard for open-source, commercially usable Large Language Models (LLMs) and have been downloaded over 3 million times. MosaicML firmly believes that a company's AI models are as valuable as any other core intellectual property and that high-quality AI models should be accessible to everyone. Being a part of Databricks since July 2023, we are enthusiastic about helping our customers solve the world's most challenging problems - from establishing the next mode of transportation to accelerating the advancement of medical breakthroughs. We manage the world's best data and AI platform, enabling our customers to use profound data insights to improve their businesses. We seize every opportunity to solve technical issues, aiming to equip our customers with the best data and AI abilities. Summary: MosaicML is hiring experienced machine learning platform engineers to build our generative AI platform, covering the ML model development lifecycle, which involves pre-training of large language models (LLMs), fine-tuning, evaluation, and serving. You'll thrive in this position if you possess a strong sense of overall responsibility and take pleasure in converting user requirements into product interfaces while building the backend distributed systems to power these interfaces. You will have the chance to contribute to all aspects of our stack, including our CLI and SDK interfaces, backend APIs, distributed orchestration systems built on Kubernetes, and core platform infrastructure. Responsibilities: - Play a major role in the design and implementation of our product, a platform for supporting use cases across the training and deployment of generative AI models. - Collaborate closely with both ML researchers and clients to identify key development areas for our generative AI platform. - Show strong product ownership, translating product requirements into user interfaces and backend distributed system design. - Design and establish the core platform infrastructure supporting our client-facing product features. - Guarantee the reliability, security, and scalability of the backend distributed systems that power all aspects of our product. Qualifications: - 4+ years of full-time industry experience. - Strong software engineering skills. - Experience with large-scale distributed systems. - Experience building ML platform systems for applications in the ML model development lifecycle. - Direct experience developing ML models is a plus but not required. - Strong sense of product ownership and intuition for both robust system design and product usability. - Effective communication skills and the ability to simplify complex technical concepts to diverse internal and external stakeholders. We appreciate candidates who show curiosity about all parts of the company's success and are eager to learn new technologies. Salary Range: Databricks is devoted to fair and equal compensation practices. The pay range for this role is $166,000-$225,000 USD, representing the base salary for non-commissionable roles or on-target earnings for commissionable roles. Actual compensation packages are determined by various factors unique to each candidate, such as job-related skills, experience, relevant certifications and training, and specific work location. About Databricks: Databricks is the data and AI company. More than 9,000 organizations worldwide, including Comcast, Condé Nast, and over 50% of the Fortune 500, rely on the Databricks Lakehouse Platform to unify their data, analytics, and AI. Databricks is headquartered in San Francisco, with offices around the world. Founded by the original creators of Apache Spark™, Delta Lake, and MLflow, Databricks is on a mission to help data teams solve the world’s toughest problems. Our Commitment to Diversity and Inclusion: At Databricks, we are dedicated to fostering a diverse and inclusive culture where everyone can excel. We ensure that our hiring practices are inclusive and meet equal employment opportunity standards. Candidates seeking employment at Databricks are considered without discrimination. Compliance: If access to export-controlled technology or source code is required to perform the job.