JAX Jobs in the East of England

6 of 6 JAX Jobs in the East of England

Senior Software Engineer - Machine Learning Tools

Cambridge, Cambridgeshire, United Kingdom
Hybrid / WFH Options
NLP PEOPLE
we actively support that growth. • Experience shipping real products to customers • ML Design and Deployment: Familiarity with designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT, HuggingFace Transformers, or commercial APIs. • Containerisation: Experience with cloud deployments and containerisation (Docker, Kubernetes, containerd). • Linux: Embedded Linux and kernel development. • Programming Languages: We adopt a flexible tooling 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:

Senior Software Development Engineer, AWS Neuron Frameworks

Cambridge, Cambridgeshire, United Kingdom
Amazon
software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators. As a part of the Neuron Frameworks team you'll develop and enhance support PyTorch and JAX for AWS Neuron, working with the open source ecosystem. You will develop and extend support for the leading ML frameworks, delivering an outstanding user experience for PyTorch and JAX ML … will work closely with teams across AWS Neuron including compiler, training and inference optimization to optimize frameworks for AWS's accelerator architectures, and engage closely with the PyTorch and JAX and other ML Framework communities to take advantage of their latest capabilities and improve performance and usability for ML model developers. A successful candidate will have a experience developing Machine … in AWS, including support for customers who require specialized security solutions for their cloud services. Key job responsibilities You will work with the team to develop, improve and release JAX and PyTorch framework support for AWS Neuron. You will understand current and future directions of ML framework development, with a focus on setting the direction for the AWS Neuron framework 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:
JAX
the East of England
25th Percentile
£62,000
Median
£64,000
75th Percentile
£66,000