AI / ML Engineer - New Algorithms for Cryo ET

Redwood City, CA, United States

To develop a dynamic, integrated view of biological systems in health and disease, the Chan Zuckerberg Institute for Advanced Biological Imaging (CZ Imaging Institute) aims to push forward the development and application of new imaging technologies over the next 10 to 15 years. CZ Imaging Institute researchers will collaboratively develop breakthrough biological imaging systems centered around grand challenges that push the boundaries of what we can see and measure to obtain deep insights into the architecture of complex biological systems at the molecular level.

As an extension of the Chan Zuckerberg Initiative's Imaging program, the CZ Imaging Institute (https://czii.org/) will create breakthrough technologies - hardware, software, biological probes, data, and platforms - that will be made available to the scientific community and adopted worldwide through a combination of direct access to the Institute, open sharing of advances, and commercial partnerships.

The Opportunity

The CZ Imaging Institute will create breakthrough technologies - hardware, software, biological probes, data, and platforms - that will be made available to the scientific community and adopted worldwide through a combination of direct access to the institute, open sharing of advances, and commercial partnerships. Researchers will collaboratively develop breakthrough biological imaging systems centered around grand challenges that push the boundaries of what we can see and measure.

The Chan Zuckerberg Imaging Institute is seeking an experienced AI/Machine Learning Engineer to support the New Algorithms for Cryo ET Team Leader to create innovative algorithms that address specific challenges in computer vision, generating meaningful and high-quality data representations that facilitate downstream tasks.

What You'll Do

Develop robust unsupervised computer vision algorithms intended to excel in segmenting and denoising raw cryo-EM data.

Tailor established unsupervised computer vision algorithm methods to more effectively tackle the unique computer vision obstacles encountered in cryo-EM data processing.

Facilitate, manage and maintain training of large models on distributed systems.

Collaborate with individuals both within and outside of CZI.

Engage in enhancing various aspects of a large computational pipeline.

What You'll Bring

Essential-

A Master's degree in Software Engineering or Machine Learning

At least 3 years experience working with AI/ML tools and Algorithms

Have experience in representation learning techniques, including expertise in unsupervised and self-supervised learning methods, dimensionality reduction, and feature extraction and the ability to create meaningful and high-quality data representations that facilitate downstream tasks

A track record of creating innovative algorithms that address specific challenges in computer vision, inverse problems, and related fields, including a proven ability to adapt and combine existing techniques or develop novel approaches to solve complex problems

Expertise in programming languages commonly used in machine learning and computer vision, such as Python, C++, and/or CUDA with a demonstrated ability to write efficient, scalable, and well-documented code

Hands-on experience with deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) and a solid understanding of neural network architectures for tasks like image recognition, segmentation, and generative modeling

Effective teamwork and communication skills, including the ability to work collaboratively in cross-functional teams, present complex technical concepts to non-technical stakeholders, and contribute to the growth and mentorship of junior team members

Demonstrated proficiency in designing, implementing, and optimizing machine learning algorithms for various applications, with a focus on computer vision and image analysis

Proficiency in computer vision techniques, including object detection, image segmentation, feature extraction, and image classification

Developed experience in reconstructing from incomplete or noisy data (Inverse Problem) and proficiency in employing regularization techniques and optimization algorithms to stabilize and solve such problems effectively

Proficiency in implementing and adapting numerical algorithms to address challenges in machine learning and image processing

Nice to have -

A PhD in Software Engineering or Machine Learning

In-depth knowledge of advanced mathematical concepts and their practical application to real-world problems, including familiarity with linear algebra, calculus, optimization, and statistics, particularly as they pertain to machine learning and computer vision.

Strong understanding of numerical methods and their application in solving complex mathematical problems

History of research and innovation in machine learning and computer vision, demonstrated through publications, conference presentations, or project outcomes with an ability to stay updated with the latest advancements in the field

Compensation

ML Eng II = $131,000 - 180,400

ML Sr. Engineer = $150,000 - 205,700

New hires are typically hired into the lower portion of the range, enabling employee growth in the range over time. To determine starting pay, we consider multiple job-related factors including a candidate's skills, education and experience, market demand, business needs, and internal parity. We may also adjust this range in the future based on market data. Your recruiter can share more about the specific pay range during the hiring process.

What We Provide Resources to collaboratively develop breakthrough biological imaging systems that push the boundaries of what we can see and measure

An environment where nothing is considered impossible, where bold and creative thinking are the norm, and where a collaborative culture allows teams of people from multiple disciplines to accomplish more than any single person can achieve alone

Access to collaborators and resources at partner organizations in the CZ Biohub Network and beyond

Competitive compensation and benefits commensurate with experience

Benefits

We offer a robust benefits program that enables the important work employees do everyday. Our benefits include healthcare coverage, life and disability insurance, commuter subsidies, family planning services with fertility care, childcare stipend, 401(k) match, flexible time off and a generous parental leave policy. In addition, we honor our commitment to career development and our value of scholarly excellence through regular onsite opportunities to learn from the world's leading scientists.

The CZ Biohub Network is an equal opportunity employer committed to diversity of thought, ideas and perspectives. We are committed to cultivating an inclusive organization where all employees feel inspired and know their work makes an important contribution. Therefore, we provide employment opportunities without regard to age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, or any other protected status in accordance with applicable law.

Pursuant to the California Fair Chance Act, we will consider for employment qualified applicants with arrest and conviction records.

Headhunters and recruitment agencies may not submit resumes/CVs through this website or directly to managers. The CZ Biohub Network does not accept unsolicited headhunter and agency resumes. The CZ Biohub Network will not pay fees to any third-party agency or company that does not have a signed agreement with the CZ Biohub Network.