Senior / Staff Machine Learning Research Engineer

Translate frontier machine-learning prototypes into scalable systems for drug discovery. The role optimises models such as diffusion networks, transformers and DNA-sequence models in JAX or PyTorch, improves distributed training and inference, and develops research libraries and evaluation tools.

BS/MS with at least seven years or PhD with at least four years of relevant experience, strong Python and JAX/PyTorch skills, and experience taking advanced models from concept to publication or production. Onsite at least four days per week.