Google JAX Jobs in Cambridge

6 of 6 Google JAX Jobs in Cambridge

Lead AI Scientist - Small Molecule Discovery

Cambridge, Cambridgeshire, United Kingdom
Orion Pharma
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 ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

25.05.02 Lead AI Scientist, Small Molecule Discovery, Orion Pharma, Cambridge, UK

Cambridge, Cambridgeshire, United Kingdom
The Computational Chemistry List
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 ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior AI Engineer

Cambridge, England, United Kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:

Senior AI Engineer

cambridge, east anglia, united kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:

Senior AI Engineer

Cambridge, south west england, united kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯
Posted:

Machine Learning Researcher - LLM/VLM

Cambridge, Cambridgeshire, Staines, Surrey, United Kingdom
Redline Group Ltd
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 ❯
Employment Type: Permanent
Salary: £45000 - £70000/annum
Posted: