JAX Jobs in Cambridgeshire

6 of 6 JAX Jobs in Cambridgeshire

Research Scientist

Cambridge, Cambridgeshire, Staines, Surrey, United Kingdom
Redline Group Ltd
that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C/ More ❯
Employment Type: Permanent
Salary: £30000 - £50000/annum
Posted:

Software Engineer - AI Developer Tooling

Cambridge, Cambridgeshire, United Kingdom
Hybrid / WFH Options
Arm Limited
facing developer workflows. Experience developing for and on Arm architectures. Experience profiling or debugging ML workloads. Experience designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT or HuggingFace Transformers. Experience collaborating closely with product or design teams. In Return: Our team offers an inclusive and highly collaborative environment, with work that spans a wide variety More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Staff Machine Learning Engineer - Modeling

Cambridge, Cambridgeshire, United Kingdom
Second Renaissance
inference, causal reasoning & inference, transfer & multi-task learning, diffusion models, graph neural networks, active learning; Experience writing production-quality code with modern machine learning frameworks such as PyTorch, TensorFlow, JAX, or similar; Experience with multi-GPU and distributed training at scale; A team player who thrives in collaborative environments and is committed to enabling colleagues to reach their full potential … physics, or computational biology, or equivalent experience; Experience in developing machine learning models; Very strong programming skills, including experience with Python and deep learning libraries such as PyTorch or JAX; Experience in large-scale distributed optimization of machine learning models across multiple GPUs and nodes. Preferred Qualifications - Familiarity with biological data formats, concepts, and computational models; Experience in cell health More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist - Computational Biology (Multiple Levels)

Cambridge, Cambridgeshire, United Kingdom
Deep Genomics Inc
About Us Deep Genomics is at the forefront of using artificial intelligence to transform drug discovery. Our proprietary AI platform decodes the complexity of genome biology to identify novel drug targets, mechanisms, and genetic medicines inaccessible through traditional methods. With More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

25.03.25 Machine Learning Scientists/Engineers, Cambridge, UK

Cambridge, Cambridgeshire, United Kingdom
Hybrid / WFH Options
Computational Chemistry List, Ltd. (CCL)
Machine Learning Scientists/Engineers, Cambridge, UK Astex Pharmaceuticals is a global leader in innovative drug discovery, utilizing its proprietary Fragment-Based Drug Discovery platform to develop new drug candidates, with several progressing in clinical trials. The company focuses on More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Consultant | Cambridge | Consultancy

Cambridge, Cambridgeshire, United Kingdom
SoCode Limited
About the role: Join a mission-focused technology consultancy delivering breakthrough AI and machine learning solutions to complex, real-world problems. You will apply your curiosity, creativity, and rigour to design and deploy ML systems across diverse projects, working in More ❯
Employment Type: Permanent
Posted: