Machine Learning Research Engineer Foundation Models Senior Staff Principal

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.

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

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

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