Machine Learning Engineer

Job Summary

A company is looking for a Machine Learning Engineer to develop and deploy advanced machine learning models for drug discovery.

Key Responsibilities

• Develop and train transformer-based models on large multimodal datasets
• Write modules in a PyTorch codebase for encoding and decoding chemical and drug data
• Conduct statistical analysis of model performance and deploy models in existing tools

Required Qualifications

• Fluent in PyTorch and/or JAX
• Experience with high performance modules for structured data encoders/decoders
• Proficient in manipulating data in a multi-cloud, decentralized infrastructure
• Experience with distributed model training
• Familiarity with molecular data manifolds is a plus