(Senior) Machine Learning Engineer, GPU Scaling

FL94 Inc., is a privately held, early-stage biotechnology company pioneering the new field of Protein Editing. At FL94 we create small molecules that edit protein structure and function to unlock presently undruggable targets and a broad array of therapeutic modalities.Our platform integrates novel small molecule chemistry and chemo-proteomic discovery with Machine Learning (ML) to enable generative design. FL94 is backed by Flagship Pioneering, bringing their courage, vision, and resources to guide FL94 from platform validation to patient impact. We are seeking collaborative, relentless problem solvers that share our passion for impact to join us!As a Machine Learning Engineer you will lead efforts to efficiently train massive generative models across multiple GPUs and build a world-class infrastructure that enables robust data flow between our experimental and computational platforms.Working closely with ML scientists, biologists, and chemists you will not only lead the development of this infrastructure, you will also help identify new areas across any function where ML-driven automation and workflow will impact high-quality discovery.The ideal candidate sees building robust integrated hardware/software systems for training and deploying ML models in biology as equally valuable to experimental breakthroughs and ML-focused algorithmic development.Responsibilities include:

Building scalable machine learning pipelines, including efficient parallelization and use of GPUs.Designing and coding data processing pipelines, ownership of company-wide data platform.Working closely with Platform Development and Biology teams to automate a tight design-build-test cycle for ML generated small molecules.Working in a collaborative CI/CD software development environment, including use of git, participating in code reviews, and independent development of robust code.Working within an entrepreneurial, collaborative, and interdisciplinary team to actively contribute to creating, shaping, and executing the vision of the company.Qualifications:

Undergraduate or Master's degree in bioinformatics, computer science, mathematics, physics or related field and 4+ years of experience is required.Experience with MLOps practices and tools including version control, automated testing, and CI/CD.Proficient in Python and experience in using deep learning libraries (i.e. PyTorch).Experience with cloud computing (i.e. AWS), database scale-out (e.g. distributed postgresql), and data pipelining tools (i.e. Nextflow, Airflow).Experience with Slurm or other on-site HPC cluster solutions.Nice-to-have, but not critical, is experience with computational chemistry or protein structure platforms and collaborative tools.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.At Flagship, we recognize there is no perfect candidate. If you have some of the experience listed above but not all, please apply anyway. Experience comes in many forms, skills are transferable, and passion goes a long way. We are dedicated to building diverse and inclusive teams and look forward to learning more about your unique background.

#J-18808-Ljbffr