JAX Job Vacancies

26 to 50 of 51 JAX Jobs

Research Scientist

Peterborough, Cambridgeshire, United Kingdom
microTECH Global Limited
experience in academia or industry • We will consider various levels of experience in relevant research areas Key Skills: • Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) • Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist

Peterborough, Cambridgeshire, United Kingdom
Hyper Recruitment Solutions Ltd
machine learning, including deep learning and/or generative models. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling/ More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist

Cambridgeshire, England, United Kingdom
Hyper Recruitment Solutions
learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling/ More ❯
Posted:

Machine Learning Scientist

cambridgeshire, east anglia, united kingdom
Hyper Recruitment Solutions
learning, including deep learning and/or generative models. 3. A working knowledge and practical experience with modern ML frameworks (e.g., PyTorch, TensorFlow, or JAX). Key Words: Machine Learning/Drug Discovery/Computational Chemistry/Structural Biology/AI Techniques/Protein-Ligand Interaction/Molecular Modelling/ More ❯
Posted:

AI/ML Engineer, National Security, Public Sector with Security Clearance

Reston, Virginia, United States
Google, Inc
Science, a related technical field, or equivalent practical experience. 6 years of experience with machine learning model development and deployment, frameworks (e.g., PyTorch, Tensorflow, Jax, Ray, etc.), AI accelerators (e.g., TPUs, GPUs), model architectures (e.g., encoders, decoders, transformers), and using machine learning Application Programming Interface (APIs). 6 years of More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Artificial Intelligence Engineer

United Kingdom
NineTech
Key Tech the client are using: 🧠 LLMs (GPT, Claude, LLaMA, etc.) 🛠️ LangChain, AutoGPT, CrewAI 🤖 Reinforcement Learning & Multi-Agent Systems 📡 Kafka, RabbitMQ, OpenTelemetry 🧰 TensorFlow, PyTorch, JAX ☁️ AWS, Azure, Vertex AI What You'll Do: • Build and deploy autonomous agents • Optimize real-time AI models • Implement RL and planning algorithms • Integrate with More ❯
Posted:

Principal Solutions Architect, Generative AI, Startups, AWS

London, United Kingdom
ENGINEERINGUK
and deploying GenAI models and applications on cloud infrastructure. Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. Knowledge of MLOps tools and workflows for model development, validation, and deployment. High level of communicating effectively across internal and external organizations, with More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Principal Solutions Architect, Generative AI, Startups, AWS

London, United Kingdom
Amazon
and deploying GenAI models and applications on cloud infrastructure. - Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. - Knowledge of MLOps tools and workflows for model development, validation, and deployment. - High level of communicating effectively across internal and external organizations, with More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Head of Applied AI

London Area, United Kingdom
Hybrid / WFH Options
Acquired Talent Ltd
collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology in More ❯
Posted:

Head of Applied AI

london, south east england, united kingdom
Hybrid / WFH Options
Acquired Talent Ltd
collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology in More ❯
Posted:

Senior AI Engineer

Cambridge, England, United Kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and/ More ❯
Posted:

Senior AI Engineer

cambridge, east anglia, united kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and/ More ❯
Posted:

Senior AI Engineer

Cambridge, south west england, united kingdom
DeepRec.ai
Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and/ More ❯
Posted:

Member of Technical Staff, Training Infra Engineer

London, United Kingdom
Cohere
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 with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray). Experience using large-scale distributed training strategies. Hands More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Data Scientist, AWS Generative AI Innovation Center

London, United Kingdom
Amazon
Data Scientist, AWS Generative AI Innovation Center Job ID: Amazon Web Services EMEA Dubai FZ Branch - Q29 Amazon launched the Generative AI Innovation Center (GenAIIC) in June 2023 to help AWS customers accelerate the use of generative AI to solve More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior AI Scientist

Diepenbeek, Limburg, Belgium
Rejuvenate Biomed
Rejuvenate Biomed is a clinical-stage biotechnology company based in Diepenbeek, Belgium. Our mission is to discover and develop therapeutics to delay the onset or progression of age-related diseases. We are looking for an experienced and innovative Senior AI More ❯
Employment Type: Permanent
Salary: EUR Annual
Posted:

Machine Learning Engineer

London, United Kingdom
Graphcore
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 Engineer, AI for Electrical Power Systems

london, south east england, united kingdom
DeepMind
Snapshot Artificial Intelligence could be one of humanity's most useful inventions. At Google DeepMind, we're a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We More ❯
Posted:

Artificial Intelligence Engineer

London, UK
Understanding Recruitment
AI Research Engineer (Remote) A leading defence tech startup building AI and software for autonomous systems is seeking an AI Research Engineer to join their remote team. This role, managed by a third-party recruitment agency, offers the chance to More ❯
Posted:

Machine Learning Researcher - LLM/VLM

Surrey, United Kingdom
Redline Group Ltd
Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯
Employment Type: Permanent
Salary: GBP 45,000 - 70,000 Annual
Posted:

Machine Learning Researcher - LLM/VLM

Cambridge, Cambridgeshire, Staines, Surrey, United Kingdom
Redline Group Ltd
Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯
Employment Type: Permanent
Salary: £45000 - £70000/annum
Posted:

Machine Learning Researcher - LLM/VLM

Staines, Surrey, South East, United Kingdom
Redline Group Ltd
Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯
Employment Type: Permanent
Salary: £70,000
Posted:

Machine Learning Engineer

London Area, United Kingdom
GCS AI
Machine Learning Engineer | Research Platform: We are a tier 1 supplier for a well-known buy-side quant firm on the lookout for a Machine Learning Engineer (£475,000-£515,000) who is happy to work at the interface of More ❯
Posted:
JAX
10th Percentile
£69,357
25th Percentile
£80,393
Median
£88,863
75th Percentile
£93,401
90th Percentile
£97,360