Machine Learning Scientist, Pretraining

Machine Learning Scientist, PretrainingEmeryville, California, United States; Hybrid (2-3 days on-site)Profluent is the frontier AI lab for biology. Profluent builds powerful foundation models for all of life's molecules, unlocking solutions that transform medicine, agriculture, and beyond. Founded in 2022 and headquartered in Emeryville, CA, Profluent is backed by leading investors and has raised over $150M to date.The RoleWe're looking for a motivated and creative Machine Learning (ML) Scientist to drive research into pretraining large-scale deep learning models for biomolecular design. This position offers an opportunity to work at the forefront of generative modeling research across language processing, representation learning, and protein engineering. You should be a self-directed researcher who has the ability to rapidly prototype and evaluate new models and algorithms in the biomolecular domain.As an early employee, you will proactively shape the direction of our machine learning efforts and collaborate across diverse teams of computational and experimental scientists.ResponsibilitiesDesign and develop state-of-the-art autoregressive, diffusion and representation learning models for protein designCollaborate across the machine learning and protein design teams to apply, adapt and improve pretraining techniques from other domains to biomolecular deep learning modelsArchitect, implement, and optimize core infrastructure to support the pretraining of protein language modelsCurate relevant datasets and design tasks for rigorous evaluation of generative modelsImplement, analyze, and interpret multiple computational approaches and present results to colleagues in regular update meetingsWork within a collaborative, fast-paced, interdisciplinary team across biology and machine learning to help shape the scientific and strategic vision of the companyQualificationsPhD (or equivalent industry experience) in Computer Science, Machine Learning, Natural Language Processing, Applied Math, Computational Biology, Statistics, or a related fieldExperience with conceiving of, implementing, and evaluating novel machine learning and pretraining large scale LLMs or other models in biomolecular domainPublications at major machine learning conferences (NeurIPS, ICML, ICLR) or scientific journals (Nature, Science, Nature Biotech, Nature Methods, PNAS) or experience pre-training LLMs at frontier AI/ML labsExperience with modern deep learning frameworks such as Pytorch or JaxPreferencesFamiliarity with foundational biology of proteins and nucleic acidsExperience developing machine learning models for proteins (language models, structure prediction, design)Experience with cloud compute platforms (GCP, AWS, Azure, OCI)Previous experience in data extraction and curation from bioinformatics data sourcesFamiliarity with wet lab experimental assays and associated limitationsWhat We OfferHigh-growth opportunity with meaningful impact on the future of protein designCompetitive compensation package with equity participation401(k) with a strong employer matchComprehensive benefits including health/dental/vision insuranceGenerous PTO policy and commitment to work-life balanceProfessional development opportunities in a cutting-edge field at the intersection of AI and biologyProfluent Bio, Inc is an equal opportunity employer promoting diversity and inclusion in the workspace. We do not discriminate on the basis of race, color, religion, marital status, age, national origin, ancestry, physical or mental disability, medical conditions, veteran status, sexual orientation, gender (including gender identity and gender expression), sex (which includes pregnancy, childbirth, and breastfeeding), genetic information, taking or requesting statutorily protected leave, or any other basis protected by law.Legal authorization to work in the United States is required. In compliance with federal law, all persons hired must verify their identity and work eligibility and complete the required employment verification form upon hire.Hiring Salary Range $200,000 - $330,000 USD