Permanent Google JAX Jobs in England

1 to 25 of 45 Permanent Google JAX Jobs in England

Senior AI Engineer

England, United Kingdom
Scrumconnect Limited
the team grows. Key Responsibilities Implement end-to-end AI pipelines: data collection/cleaning, feature engineering, model training, validation, and inference. Rapidly prototype novel models using PyTorch, TensorFlow, JAX, or equivalent. Productionize models in cloud/on-prem environments (AWS/GCP/Azure) with containerization (Docker/Kubernetes). 2. Data & Infrastructure Build and maintain scalable data More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer (London)

London, UK
Hybrid / WFH Options
InstaDeep Ltd
at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … 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 More ❯
Employment Type: Full-time
Posted:

Research Engineer, Machine Learning (London)

London, UK
Hybrid / WFH Options
Mistral AI
Join to apply for the Research Engineer, Machine Learning role at Mistral AI Continue with Google Continue with Google Join to apply for the Research Engineer, Machine Learning role at Mistral AI About Mistral At Mistral AI, we believe in the power of AI to simplify tasks, save time, and enhance learning and creativity. Our technology is designed … enterprise API About You Master’s or PhD in Computer Science (or equivalent proven track record) 4 + years working on large-scale ML codebases Hands-on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed/FSDP/SLURM/K8s) Experience in deep learning, NLP or LLMs; bonus for CUDA or data-pipeline chops Strong software … Technology Industries Technology, Information and Internet Referrals increase your chances of interviewing at Mistral AI by 2x Sign in to set job alerts for “Research Engineer” roles. Continue with Google Continue with Google Continue with Google Continue with Google London, England, United Kingdom 3 weeks ago Hampton, England, United Kingdom 6 days ago Quantum Algorithms Scientist More ❯
Employment Type: Full-time
Posted:

AI Security Architect (IT) / Contractor

Reading, Berkshire, United Kingdom
Onyx-Conseil
Python, R, C, C++, or Java, and their associated security considerations. • Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. • Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. • Strong understanding of More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer (London)

London, UK
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: Full-time
Posted:

Machine Learning Engineer

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:

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 … 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 More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer (London)

London, UK
Graphcore
Join to apply for the Machine Learning Engineer role at Graphcore Join to apply for the Machine Learning Engineer role at Graphcore Get AI-powered advice on this job and more exclusive features. As a Machine Learning Engineer in the More ❯
Employment Type: Full-time
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

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:

Research Engineer, Machine Learning (Horizons) (London)

London, UK
Hybrid / WFH Options
Menlo Ventures
may be a good fit if you: Have 5+ years of industry-related experience Are proficient in Python and have experience with deep learning frameworks such as PyTorch or Jax Have a strong software engineering background and are interested in working closely with researchers and other engineers Enjoy pair programming (we love to pair!) Care about code quality, testing … file types: pdf, doc, docx, txt, rtf Enter manually Accepted file types: pdf, doc, docx, txt, rtf (Optional) Personal Preferences * How do you pronounce your name? Website Publications (e.g. Google Scholar) URL Other Useful URLs (e.g. Blog Posts) Are you open to working in-person in one of our offices 25% of the time? * Select... Of the locations listed More ❯
Employment Type: Full-time
Posted:

Research Engineer, Machine Learning (Horizons)

London, United Kingdom
Hybrid / WFH Options
Menlo Ventures
may be a good fit if you: Have 5+ years of industry-related experience Are proficient in Python and have experience with deep learning frameworks such as PyTorch or Jax Have a strong software engineering background and are interested in working closely with researchers and other engineers Enjoy pair programming (we love to pair!) Care about code quality, testing … file types: pdf, doc, docx, txt, rtf Enter manually Accepted file types: pdf, doc, docx, txt, rtf (Optional) Personal Preferences How do you pronounce your name? Website Publications (e.g. Google Scholar) URL Other Useful URLs (e.g. Blog Posts) Are you open to working in-person in one of our offices 25% of the time? Select Of the locations listed 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:

AI Research Scientist (AEC)

London, United Kingdom
Merantix
and deploying production-ready solutions) Familiarity in statistical methods for Machine Learning (e.g. Bayesian methods, HMMs, graphical models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Strong coding abilities in Python Preferred Qualifications Experience in the Architecture, Engineering, and/or Construction domains, including expertise with industry-specific data formats (e.g., BIM More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI Research Scientist

Cambridge, Cambridgeshire, United Kingdom
Hybrid / WFH Options
microTECH Global Limited
Permanent Role Location : Cambridge Hybrid Policy: 3 days onsite and 2 days working from home weekly Role and Responsibilities As anAI Research Scientist, you will be part of a dynamic team working on next-generation machine learning technologies for real 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:

Research Scientist

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

London, UK
Woven by Toyota
Join to apply for the Machine Learning Engineer role at Woven by Toyota Continue with Google Continue with Google Join to apply for the Machine Learning Engineer role at Woven by Toyota Woven by Toyota is enabling Toyota’s once-in-a-century transformation into a mobility company. Inspired by a legacy of innovating for the benefit of … and requirements for an end-to-end stack; Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state-of-the-art architectures for end-to-end autonomous driving … Technology Industries Software Development Referrals increase your chances of interviewing at Woven by Toyota by 2x Sign in to set job alerts for “Machine Learning Engineer” roles. Continue with Google Continue with Google Continue with Google Continue with Google London, England, United Kingdom 2 weeks ago London, England, United Kingdom 1 week ago London, England, United More ❯
Employment Type: Full-time
Posted:

Research Scientist

Staines-upon-Thames, Middlesex, England, United Kingdom
Redline Group Ltd
that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C More ❯
Employment Type: Full-Time
Salary: £30,000 - £50,000 per annum
Posted:

Research Scientist

Staines, Surrey, South East, United Kingdom
Redline Group Ltd
that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C More ❯
Employment Type: Permanent
Salary: £50,000
Posted:

Research Scientist

Cambridge, Cambridgeshire, Staines, Surrey, United Kingdom
Redline Group Ltd
that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C More ❯
Employment Type: Permanent
Salary: £30000 - £50000/annum
Posted:

Research Engineer

London, United Kingdom
Deepstreamtech
problems. (Desirable) Experience designing and conducting experiments in a social science or human-computer interaction (HCI) context. Strong proficiency in Python and common ML libraries/frameworks (e.g., PyTorch, JAX, TensorFlow, Hugging Face). (Desirable) A track record of publications in relevant AI or interdisciplinary conferences. A strong scientific mindset geared towards rigorous experimentation, coupled with the engineering discipline More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Head of AI R&D (London)

London, UK
Kraken
models, AI agent orchestration, benchmarking, and deployment. Proven experience in AI research and leading complex projects. Experience with data enrichment and knowledge base development. Proficiency in PyTorch, TensorFlow, or Jax, and programming languages like Python, R, Java. Preferred: publications in peer-reviewed journals and conferences. Preferred: experience in the energy sector. If this sounds like you, we'd love More ❯
Employment Type: Full-time
Posted:

Hewlett Packard Labs - Research Engineer - MINDnet

Bristol, Gloucestershire, United Kingdom
Hewlett Packard Enterprise Development LP
self-motivated and independent worker. Strong English writing and verbal communication skills. Preferred Skills Proficiency in Python. Expertise in one of the standard machine learning frameworks, such as PyTorch, JAX, or TensorFlow. Broad knowledge of and practical experience with modern machine learning and optimization techniques. Excellent analytical, numerical, and modeling skills. Broad knowledge of photonics and general physics. Familiarity 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 Anthropic’s 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 More ❯
Employment Type: Full-time
Posted:
Google JAX
England
10th Percentile
£61,000
25th Percentile
£65,250
Median
£85,000
75th Percentile
£97,500