JAX Jobs in England

26 to 34 of 34 JAX Jobs in England

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:

Member of Technical Staff, Training Performance Engineer

London, United Kingdom
Cohere
the best researchers in the field. You may be a good fit if you have: Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR. Experience writing kernels for GPUs using CUDA, triton, etc Experience using large-scale distributed training strategies. Familiarity with autoregressive sequence models, such as Transformers. Bonus 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 Engineer (Large Systems)

Bristol, Gloucestershire, 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 Engineer (Large Systems)

London, 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:

Research Scientist

London, United Kingdom
Cerebras
Job Summary As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of 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:

Senior Applied Scientist

London, United Kingdom
Unlikely Artificial Intelligence Limited
At UnlikelyAI, we're looking for a Senior Applied Scientist to join our Applied Science team. This is a high-impact individual contributor position. You'll help drive the end-to-end lifecycle of projects: from identifying opportunities in literature More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Staff Applied Scientist

London, United Kingdom
Unlikely Artificial Intelligence Limited
At UnlikelyAI, we're looking for a visionary Staff Applied Scientist to play a key leadership role in our Applied Science team and across the company. This is a high-impact individual contributor position with mentoring and coaching responsibilities - ideal More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:
JAX
England
10th Percentile
£61,000
25th Percentile
£65,250
Median
£85,000
75th Percentile
£97,500