Permanent JAX Jobs in England

11 of 11 Permanent JAX Jobs in England

AI Solutions Engineer

London, United Kingdom
Bioptimus
deep experience with the AWS ecosystem and services relevant to ML, such as SageMaker, EC2, S3, IAM, and VPC. You are familiar with deep-learning frameworks like PyTorch or JAX, and libraries from the AI ecosystem like Hugging Face Transformers. You have worked with the lifecycle of AI models, including deployment, hosting, and optimization for low-latency inference. You have More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Software Engineer, AI Fitness, Health Coach, Fitbit

London, United Kingdom
Google Inc
years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture. Experience using machine learning frameworks (e.g., Tensorflow, PyTorch, or Jax). Experience with machine learning model pretraining, fine-tuning, evaluation, and methodologies. Preferred qualifications: Master's degree or PhD in Computer Science or related technical field. Experience publishing in high-quality More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Scientist, AI for Human Simulation

London, United Kingdom
Autodesk, Inc
learning and generative AI techniques, including implementing custom architectures, pretraining/fine-tuning generative models, building agentic workflows, and deploying solutions with real-world constraints. Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Preferred Qualifications Experience in building AI agents (based on RL, LLMs, etc.) Experience in building multi-agent simulation Background in Large Language Models/NLP and Multi More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Scientist, Generative AI (Multimodal Foundation Models)

London, United Kingdom
Hybrid / WFH Options
InstaDeep Ltd
of high-quality research, evidenced by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. Strong analytical and critical thinking skills More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Applied Scientist Applied Science

London, United Kingdom
Hybrid / WFH Options
Unlikely Artificial Intelligence Limited
drive: Deep learning expertise with 2+ years in language generation. Experience with LLMs, neurosymbolic integration, and knowledge representation. Strong Python skills and experience with ML frameworks (PyTorch, HF, TensorFlow, JAX). Demonstrable track record of end-to-end applied research. Excellent communication skills; proven mentoring ability. Collaborative and proactive in fast-paced environments. Desirable: Experience with symbolic reasoning engines. Technical More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Quantitative Modeling, Vice President

London, United Kingdom
Hybrid / WFH Options
LGBT Great
to quickly pick up new technologies. Experience in Git, Unix. Exposure to SQL, or any high-performance computing language is a plus but not required. Exposure to PyTorch/Jax is a plus but not required. Our benefits To help you stay energized, engaged and inspired, we offer a wide range of employee benefits including: retirement investment and tools designed More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Software Engineer - AI Developer Tooling

Cambridge, Cambridgeshire, United Kingdom
Hybrid / WFH Options
Arm Limited
facing developer workflows. Experience developing for and on Arm architectures. Experience profiling or debugging ML workloads. Experience designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT or HuggingFace Transformers. Experience collaborating closely with product or design teams. In Return: Our team offers an inclusive and highly collaborative environment, with work that spans a wide variety 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:

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:

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 Applied Science

London, United Kingdom
Hybrid / WFH Options
Unlikely Artificial Intelligence Limited
Overview At UnlikelyAI, we are building the future of AI: one that is reliable, accurate and transparent. Our neurosymbolic technology harnesses the power of LLMs and generative AI, and combines it with classical symbolic technology to produce hallucination-resistant artificial More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:
JAX
England
10th Percentile
£66,000
25th Percentile
£75,000
Median
£95,000
75th Percentile
£97,500