2 of 2 Google JAX Jobs in Peterborough

experience in academia or industry • We will consider various levels of experience in relevant research areas Key Skills: • Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) • Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU More ❯

machine learning, including deep learning and/or generative models. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling More ❯