CareerPlanGet AI match score →

ML Research Engineer, London

Lausanne 💼 Full-time🗓 2026-06-24 → 2026-08-01

Core

Building and optimizing frontier AI models (Transformers, GNNs, Diffusion) to accelerate drug discovery and design novel molecules.

Role type

Senior Research Engineer (Machine Learning)

Builds

Predictive and generative AI models for rational drug design

Domain

Biopharmaceuticals / Computational Biology / AI

Deliverable

production ML models

Required skills

Deep learning frameworks (JAX, PyTorch), modern architectures (Transformers, GNNs, Diffusion), full ML lifecycle, software engineering, distributed systems training

Preferred skills

Research publications (NeurIPS, ICML, ICLR), low-level optimization (XLA, Triton, CUDA), domain knowledge in biochemistry/drug discovery, cloud infrastructure (GCP, Kubernetes)

Technologies

JAX, PyTorch, Transformers, GNNs, Diffusion models, XLA, Triton, CUDA, GCP, Kubernetes, Docker

Responsibilities

Translate research concepts into practical implementations; design and run experiments to evaluate model performance; optimize AI/ML models for inference and training; develop specialized tools and infrastructure; collaborate with scientists on foundational models.

Seniority

Senior, hands-on IC

Sourced via greenhouse · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.
Apply on Greenhouse ↗