Skip to content
NEXTMOVEFDE careers · United States

ML Research Engineer, Foundation Models (Senior / Staff / Principal)

Technologies

PyTorch · PyTorch Lightning · Ray Distributed Training · PyTorch Geometric · CUDA · Triton · TensorRT · diffusion models · reinforcement learning · LLMs

About Genesis Molecular AI

Builds AI-and-physics foundation models to discover and optimize small-molecule drugs for its own pipeline and pharma partners.

Series B

Job description

The full responsibilities and requirements are on the employer’s site.

Open application page
Source and classification

Internal deployment & tooling · Evidence for this classification:

computational chemistry, requiring deep technical rigor and strong interdisciplinary collaboration. About the Role This role is for a highly skilled ML Research Engineer who thrives at the intersection of fundamental research and production-grade engineering. As a core member of the Genesis AI team, you will serve as the engineering pillar for inventing, scaling, and shipping our next generation of foundation models for molecular science. You will partner closely with ML researchers, computational chemists, and drug discovery scientists to translate cutting-edge model ideas into systems that power real drug discovery programs. Your work may involve: Scaling model pretraining pipelines Advancing reinforcement learning or post-training systems Optimizing performance of large molecular models Bringing structure prediction models like Pearl into production environments used by
More from the job description

ML Research Engineer, Foundation Models About the Team Join a world-class team at the forefront of AI and biochemistry. At Genesis Molecular AI, we’re a tight-knit team of proven deep learning researchers, software engineers, and drug discovery pioneers. Our shared mission is nothing short of revolutionary: to forge the next generation of AI foundation models that unlock new therapies for patients with severe diseases. We conduct fundamental research at the intersection of machine learning, physics, and computational chemistry, pushing the boundaries of each field. The Genesis AI team is building the engine for this revolution. We develop large-scale generative models trained across the full spectrum of molecular data, supported by extensive compute infrastructure and simulation pipelines. The work sits at the intersection of machine learning research, structural biology, and computational chemistry, requiring deep technical rigor and strong interdisciplinary collaboration. About the Role This role is for a highly skilled ML Research Engineer who thrives at the intersection of fundamental research and production-grade engineering. As a core member of the Genesis AI team, you will serve as the engineering pillar for inventing, scaling, and shipping our next generation of foundation models for molecular science. You will partner closely with ML researchers, computational [... source excerpt omitted ...] aining pipelines Advancing reinforcement learning or post-training systems Optimizing performance of large molecular models Bringing structure prediction models like Pearl into production environments used by chemists and drug programs This role requires someone who can bridge ML and computational chemistry, translating between disciplines and helping teams move quickly from research insight to deployed capability. We are looking for someone who can own problems end-to-end, in a fast-moving research environment, translating novel ML ideas into systems that scientists can use in active discovery programs. Positions are available at various levels of seniority: Senior, Staff, a [... source excerpt omitted ...] s. Make cutting-edge foundation model research a reality at scale. Implement, optimize, and build novel foundation models from the initial research prototypes to high-performance production models. Optimize performance of large-scale ML systems, including distributed training, inference efficiency, and GPU-level optimizations where necessary. Constantly engage with deep learning literature, building upon novel architectures and training methods to create new capabilities. Bridge machine learning research and computational chemistry workflows, working closely with computational chemists, structural biologists, and medicinal chemists to ensure models translate effectively into re

How jobs are selected

Employer postings · Data from · Sources