Machine Learning Research Engineer Foundation Models Senior Staff Principal
AI-focused biotechnology company developing molecular-AI systems and small-molecule drug candidates for internal programs and pharmaceutical partners.
Funding history
About Genesis Molecular AI
Genesis Molecular AI combines foundation models, generative and predictive AI, and physics-based computation in its GEMS platform for small- and medium-size-molecule drug discovery.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will lead engineering and experimentation for molecular foundation models, from research prototypes to production systems. You will optimize distributed training, inference, and GPU performance; run and analyze experiments; integrate models into drug-discovery workflows; mentor colleagues; and contribute research to the wider ML community.
Requirements
- Machine learning systems
- Foundation model
- Pretraining
- Post-training
- Python
- PyTorch
- Distributed training
- Large-scale dataset
- GPU optimization
- CUDA
- PyTorch Lightning
- Ray Distributed Training
- PyTorch Geometric
- Computational chemistry
- Structural biology
- Medicinal chemistry
- Communication
Responsibilities
- Drive R&D and scaling for foundation models
- Implement and optimize foundation models from prototype through production
- Optimize distributed training, inference, and GPU performance
- Apply deep learning research to new model capabilities
- Bridge ML research and computational chemistry workflows
- Productionize structure-prediction models
- Design, execute, and analyze experiments
- Ship models to production
- Integrate models into the drug discovery platform
- Mentor researchers and engineers
- Publish research and represent the company at AI and ML conferences
Benefits
- Equity
- Medical insurance
- Dental insurance
- Vision insurance
- 401(k) plan
- Unlimited PTO
- Free office lunches and dinners
- Paid family leave
- Life insurance
- Long-term disability insurance
- Short-term disability insurance