Staff AI/ML Research Engineer (GenAI)
Description For Staff AI/ML Research Engineer (GenAI)
The Chan Zuckerberg Initiative (CZI) is seeking a Staff AI/ML Research Engineer specializing in Generative AI to join their cutting-edge team. Founded by Priscilla Chan and Mark Zuckerberg, CZI aims to solve society's toughest challenges, with a focus on accelerating biomedical science through innovative technology.
As a Staff AI/ML Research Engineer, you'll be at the forefront of applying state-of-the-art AI and machine learning models to critical problems in biomedical sciences. You'll work on one of the largest computing systems dedicated to nonprofit life science research, developing predictive models of healthy and diseased cells that could lead to groundbreaking discoveries in disease prevention and management.
Your role will involve collaborating with a diverse team of scientists, computational biologists, and engineers both within CZI and external partners. You'll be responsible for building, deploying, and maintaining large-scale machine learning models and AI agents for scientific discovery. This position offers a unique opportunity to work with CZI's impressive 1000xH100 GPU compute cluster, allowing you to push the boundaries of what's possible in AI for science.
Key responsibilities include optimizing and deploying models for both batch and real-time inference, working with multimodal biological data types, and serving as a crucial interface between research and product teams. You'll also have the chance to contribute to open science practices, furthering CZI's mission of accelerating scientific progress.
The ideal candidate will have a strong background in computer science with a focus on machine learning, demonstrated expertise in distributed computing frameworks, and experience with large-scale AI model development. You should be comfortable working in a highly collaborative, cross-functional environment and have excellent communication skills.
CZI offers a competitive salary range of $214,000 - $321,000, along with comprehensive benefits including 401(k) matching, flexible spending options, and various insurance coverages. This hybrid role is based in the San Francisco Bay Area, offering the flexibility of remote work combined with on-site collaboration.
Join CZI in their mission to build a more inclusive, just, and healthy future for everyone by applying your AI expertise to some of the most pressing challenges in biomedical science.
Responsibilities For Staff AI/ML Research Engineer (GenAI)
- Work with Research Scientists to iterate on, optimize, deploy, and maintain large-scale machine learning models
- Build, train, and evaluate multimodal foundation models utilizing CZI's 1000xH100 GPU compute cluster
- Work with diverse range of multimodal and biological data types
- Serve as an interface to product teams
- Interact with the community through open science practices
Requirements For Staff AI/ML Research Engineer (GenAI)
- Masters or Ph.D in computer science with a focus on machine learning & data analytics, or equivalent industry experience
- 3-5 years of experience developing and applying machine learning methods
- Expertise in using modern frameworks for distributed computing and infrastructure management
- Experience in fine-tuning and/or pre-training large models
- Knowledge of deploying models for both batch and real-time (endpoint) inference
- Good working knowledge of Python-based ML libraries and frameworks
Benefits For Staff AI/ML Research Engineer (GenAI)
- CZI provides a generous employer match on employee 401(k) contributions
- Annual benefit for employees that can be used for housing, student loan repayment, childcare, commuter costs, or other life needs
- CZI Life of Service Gifts
- Paid time off to volunteer
- Funding for select family-forming benefits
- Relocation support
