ML Research Engineer, Foundation Models

About the Role

• Drive the R&D and scaling of our foundation models, taking ownership of the engineering and experimentation for key research initiatives.
• 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. You will constantly engage with deep learning literature, building upon novel architectures and training methods to create new capabilities.
• Own the experimental lifecycle with scientific rigor. You'll design experimental plans, own their execution on our large-scale compute infrastructure, and drive the deep analysis of results to inform the next research cycle and to validate most promising approaches.
• Engineer our models for state-of-the-art performance, optimizing the scalability and efficiency of every part of the training and inference pipeline.
• Ship state-of-the-art models to production, working closely with the broader team to integrate your models into our drug discovery platform.