Remote JAX Jobs in England

10 of 10 Remote JAX Jobs in England

Artificial Intelligence Engineer

Greater London, England, United Kingdom
Hybrid / WFH Options
Intellect Group
Science, Machine Learning, Mathematics, or a closely related field. Strong foundational knowledge of machine learning and deep learning , with hands-on experience using frameworks such as PyTorch , TensorFlow , or JAX . Exposure to LLM fine-tuning and/or experience working with transformer-based models (e.g. GPT, LLaMA, Mistral). Interest in or familiarity with agentic AI frameworks and multi More ❯
Posted:

Senior Machine Learning Engineer

London, United Kingdom
Hybrid / WFH Options
InstaDeep Ltd
or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to thrive More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Data Scientist

London Area, United Kingdom
Hybrid / WFH Options
Kantar Media
learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or PhD More ❯
Posted:

Data Scientist

City of London, London, United Kingdom
Hybrid / WFH Options
Kantar Media
learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or PhD More ❯
Posted:

Research Scientist, LLM Agents (Foundational Research)

London, United Kingdom
Hybrid / WFH Options
Thomas Reuters
e.g., NeurIPS, ICML, ICLR, ACL,EMNLP, NAACL, ICLR) with specific focus on agent systems, tool use, or multi-agentcoordination Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Excellent communication skills to report and present research findings anddevelopments clearly, both orally and in writing. Curious and innovative disposition capable of devising novel, well-founded algorithmicsolutions to relevant More ❯
Employment Type: Permanent
Salary: GBP Annual
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:

Research Engineer, Machine Learning (Horizons) London, UK (London)

London, UK
Hybrid / WFH Options
Alcides Fonseca
Research Engineer, Machine Learning (Horizons) London, UK About Anthropic Anthropics mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is More ❯
Employment Type: Full-time
Posted:

25.03.25 Machine Learning Scientists/Engineers, Cambridge, UK

Cambridge, Cambridgeshire, United Kingdom
Hybrid / WFH Options
Computational Chemistry List, Ltd. (CCL)
Machine Learning Scientists/Engineers, Cambridge, UK Astex Pharmaceuticals is a global leader in innovative drug discovery, utilizing its proprietary Fragment-Based Drug Discovery platform to develop new drug candidates, with several progressing in clinical trials. The company focuses on More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Remote Machine Learning Compiler Engineer - Gensyn (London)

Surbiton, Greater London, UK
Hybrid / WFH Options
Blockchain Works
The world will be unrecognisable in 5 years. Machine learning models are driving our cars , testing our eyesight , detecting our cancer , giving sight to the blind , giving speech to the mute , and dictating what we consume, enjoy, and think . More ❯
Employment Type: Full-time
Posted:

Remote Machine Learning Compiler Engineer - Gensyn

London, United Kingdom
Hybrid / WFH Options
Blockchain Works
The world will be unrecognisable in 5 years. Machine learning models are driving our cars , testing our eyesight , detecting our cancer , giving sight to the blind , giving speech to the mute , and dictating what we consume, enjoy, and think . 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