3 of 3 Google JAX Jobs in Cambridge

e.g., generative models, GNNs, active learning, foundation models) in a scientific context. Strong coding skills and experience with relevant ML frameworks (e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit). Deep understanding of molecular representation techniques and their application in modeling. Experience in designing and implementing robust, scalable More ❯

e.g., generative models, GNNs, active learning, foundation models)in a scientific context.- Strong coding skills and experience with relevant ML frameworks(e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit).- Deep understanding of molecular representation techniques and their application in modeling.- Experience in designing and implementing robust, scalable More ❯

Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯