(Senior) ML Research Engineer (Generative Modeling)

Key Skills

Aartificial intelligence, Apache Airflow, AWS, ETL frameworks, GANs, Google Cloud Platform (GCP), Large Language Models - LLMs, Machine learning techniques, MLOps tools, PyG- PyTorch Geometric, Python Programming, PyTorch

Job Description

About us

What if we could build an AI engine that transforms the way we conceive and create life science innovations?

At Flagship Pioneering, we conceive, create, resource, and develop first-in-category life sciences companies to transform human health and sustainability. We’ve created over 100 scientific ventures, including the now familiar drug and vaccine innovator, Moderna Therapeutics.

Pioneering Intelligence (PI) is an initiative focused on AI/ML based scientific innovation within Flagship. Within PI, our Labs group builds new and experimental AI/ML models with the goal of creating platforms that can accelerate scientific discovery. We are a team of machine learning scientists and engineers working within an ecosystem of domain experts and entrepreneurs. We tackle foundational ML challenges and apply those solutions to address core problems in life sciences and beyond.

Project Snapshot: AI Biologist

What if we could build an AI Biologist with:

• Near-human reasoning over scientific knowledge
• Super-human ability to process and integrate multiple data modalities
• Super-human ability to run and interpret virtual (in silico) experiments

We are building on emergent capabilities of foundation models such as multi-step reasoning (LLMs), multi-modal data integration (aligned latents), and few-shot physical property prediction (protein foundation models). Some of our efforts in this space include:

• Building lifelong-learning systems comprised of foundation models and rich feedback mechanisms, tasked with solving complex scientific problems
• Improving the underlying biological foundation models, with a specific focus on the kinds of heterogenous, multimodal data omnipresent in biology

The ML Engineer Role

As part of the PI Labs team, ML Engineers develop the infrastructure that underlies much of our work and work closely with our ML Scientists to help advance our scientific projects.

Key responsibilities

Our ML Engineers are responsible for:

• Building infrastructure to accelerate or enable ML research
• Conducting low-level research to improve the efficiency and scaling of fundamental ML tools
• Working with our ML scientists to scale and interpret experiments

Requirements

• 2+ years industry experience in ML Ops or equivalent.
• Fluency with AWS, GCP, or similar cloud-computing services.
• Fluency in python and standard ML tools (e.g. PyTorch, PyG, etc.).
• Experience with containerization and task orchestration tools (e.g. Docker, Kubernetes, Slurm)
• Motivated and team oriented, with an ability to thrive in a multidisciplinary environment.
• Ability to independently plan and implement long-term ML engineering projects, while maintaining close communication with team members.
• Excellent collaboration skills. Must be able to think independently and contribute to an active intellectual environment.

Preferred experience:

• Experience with a wide range of DL models (GANs, diffusion models, large language models, etc…), and the intricacies of managing experiments
• Familiarity with ETL pipelines for large datasets, and workflow managers (e.g. AirFlow, Prefect, etc…)
• Experience working on deep scientific problems in a group setting

Flagship Pioneering and our ecosystem companies are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status.

Recruitment & Staffing Agencies*: Flagship Pioneering and its affiliated Flagship Lab companies (collectively, “FSP”) do not accept unsolicited resumes from any source other than candidates. The submission of unsolicited resumes by recruitment or staffing agencies to FSP or its employees is strictly prohibited unless contacted directly by Flagship Pioneering’s internal Talent Acquisition team. Any resume submitted by an agency in the absence of a signed agreement will automatically become the property of FSP, and FSP will not owe any referral or other fees with respect thereto.