Permanent 'JAX' Job Vacancies

51 to 75 of 90 Permanent JAX Jobs

Senior Machine Learning Engineer - Robotics

slough, south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Machine Learning Engineer - Robotics

london (city of london), south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Research Engineer

City of London, London, United Kingdom
SR2 | Socially Responsible Recruitment | Certified B Corporation™
data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯
Posted:

Research Engineer

London Area, United Kingdom
SR2 | Socially Responsible Recruitment | Certified B Corporation™
data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯
Posted:

Research Engineer

South East, United Kingdom
SR2 | Socially Responsible Recruitment | Certified B Corporation
data curation, versioning, and reproducible experimentation. You'll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Engineer

london, south east england, united kingdom
SR2 | Socially Responsible Recruitment | Certified B Corporation™
data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯
Posted:

Research Engineer

slough, south east england, united kingdom
SR2 | Socially Responsible Recruitment | Certified B Corporation™
data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯
Posted:

Research Engineer

london (city of london), south east england, united kingdom
SR2 | Socially Responsible Recruitment | Certified B Corporation™
data curation, versioning, and reproducible experimentation. You’ll Bring Strong experience with LLMs, generative, or multimodal models , ideally involving large-scale training or evaluation. Hands-on fluency in PyTorch, JAX, or TensorFlow , with experience in DeepSpeed, Megatron, FSDP, or similar . Understanding of distributed training, parallelism strategies , and scaling laws . Familiarity with post-training methods (RLHF, RLAIF, DPO, or More ❯
Posted:

Senior Machine Learning Engineer, Scaling and Performance

London, United Kingdom
Hybrid / WFH Options
InstaDeep Ltd
tools, and hardware optimisation techniques Experience with distributed training frameworks (Ray, Dask, PyTorch Lightning, etc.) Expertise with Python and/or C/C++ Development with machine learning frameworks (JAX, Tensorflow, PyTorch etc.) Passion for profiling, identifying bottlenecks, and delivering efficient solutions. Highly Desirable Track record of successfully scaling ML models. Experience writing custom CUDA kernels or XLA operations. Understanding More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Researcher (Multimodal AI & Generative Models)

United Kingdom
Hybrid / WFH Options
Intelix.AI
record Deep expertise in diffusion models and generative AI Proven experience with text-to-image generation, multimodal AI Large-scale model training experience (distributed systems, GPUs) Proficiency in PyTorch, JAX, or similar frameworks Enterprise AI deployment experience Safety and alignment research background Multilingual/multicultural AI expertise Highly Valued: A Background at top-tier AI labs (DeepMind, OpenAI, Meta AI More ❯
Posted:

Senior Computer Vision Engineer - Robotics

City of London, London, United Kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Computer Vision Engineer - Robotics

london, south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Computer Vision Engineer - Robotics

london (city of london), south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Computer Vision Engineer - Robotics

slough, south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Software Engineer - Autonomous Robotics

London Area, United Kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Software Engineer - Autonomous Robotics

City of London, London, United Kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Software Engineer - Autonomous Robotics

South East, United Kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I've see with the More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Software Engineer - Autonomous Robotics

london, south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Software Engineer - Autonomous Robotics

london (city of london), south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Senior Software Engineer - Autonomous Robotics

slough, south east england, united kingdom
Signify Technology
looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯
Posted:

Internship in AI for Communication Systems - Students pursuing PhD only

City of London, London, United Kingdom
MediaTek
Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯
Posted:

Internship in AI for Communication Systems - Students pursuing PhD only

London Area, United Kingdom
MediaTek
Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯
Posted:

Internship in AI for Communication Systems - Students pursuing PhD only

london, south east england, united kingdom
MediaTek
Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯
Posted:

Internship in AI for Communication Systems - Students pursuing PhD only

slough, south east england, united kingdom
MediaTek
Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯
Posted:

Internship in AI for Communication Systems - Students pursuing PhD only

london (city of london), south east england, united kingdom
MediaTek
Engineering or related quantitative fields. Background in wireless communication systems, e.g., OFDM and MIMO technologies. Track record of related publications or successful projects. Proficiency in Deep Learning frameworks (PyTorch, JAX). Familiarity with advanced AI methods (e.g., transformer models). Excellent scientific communication and teamwork skills. Enthusiasm for innovative research with real-world impact. Join us to shape the future More ❯
Posted:
JAX
10th Percentile
£66,000
25th Percentile
£75,000
Median
£95,000
75th Percentile
£97,500