Back to jobs

AI / ML Engineer - New Algorithms for Cryo ET

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 ( 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

  • 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


  • 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


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.


Apply for this job


indicates a required field

,,Google Drive,or

Accepted file types: pdf, doc, docx, txt, rtf

Cover Letter
,,Google Drive,or

Accepted file types: pdf, doc, docx, txt, rtf


Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in Chan Zuckerberg Imaging Institute ’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Race & Ethnicity Definitions

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.


Voluntary Self-Identification of Disability

Form CC-305
Page 1 of 1
OMB Control Number 1250-0005
Expires 04/30/2026

Why are you being asked to complete this form?

We are a federal contractor or subcontractor. The law requires us to provide equal employment opportunity to qualified people with disabilities. We have a goal of having at least 7% of our workers as people with disabilities. The law says we must measure our progress towards this goal. To do this, we must ask applicants and employees if they have a disability or have ever had one. People can become disabled, so we need to ask this question at least every five years.

Completing this form is voluntary, and we hope that you will choose to do so. Your answer is confidential. No one who makes hiring decisions will see it. Your decision to complete the form and your answer will not harm you in any way. If you want to learn more about the law or this form, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at

How do you know if you have a disability?

A disability is a condition that substantially limits one or more of your “major life activities.” If you have or have ever had such a condition, you are a person with a disability. Disabilities include, but are not limited to:

  • Alcohol or other substance use disorder (not currently using drugs illegally)
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS
  • Blind or low vision
  • Cancer (past or present)
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or serious difficulty hearing
  • Diabetes
  • Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders
  • Epilepsy or other seizure disorder
  • Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome
  • Intellectual or developmental disability
  • Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD
  • Missing limbs or partially missing limbs
  • Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports
  • Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)
  • Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities
  • Partial or complete paralysis (any cause)
  • Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema
  • Short stature (dwarfism)
  • Traumatic brain injury

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.