DemandTrendPermanent Google JAX Jobs in Cambridgeshire

5 of 5 DemandTrendPermanent Google JAX Jobs in Cambridgeshire

Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference

Cambridge, Cambridgeshire, United Kingdom
Amazon
runtime engineers to create, build and tune distributed inference solutions with Trn1. Experience optimizing inference performance for both latency and throughput on these large models using Python, Pytorch or JAX is a must. Deepspeed and other distributed inference libraries are central to this and extending all of this for the Neuron based system is key. Key job responsibilities This More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

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 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 … 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 More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer

Cambridge, Cambridgeshire, United Kingdom
Cerebras
As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you 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: