76 to 100 of 288 JAX Jobs in England

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

with the ability to present complex ideas clearly Preferred: Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) A strong publication record in top conferences (e.g., NeurIPS, ICML, ICLR, ACL) Why Join Us Work on groundbreaking AI research with real-world applications and a strong commitment to open More ❯

have: •Experience in optimization, reinforcement learning, and/or large language models (LLMs) •Experience in chip design and applied AI research •Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Why Join Us •Work on groundbreaking AI research with real-world applications and a strong commitment to open-source contributions and scientific advancement •Collaborate with a team of world-class More ❯

look for in Research Engineers are: Degree in computer science, electrical engineering, science, mathematics or equivalent experience. Extensive software engineering experience, particularly with Python-based scientific libraries such as JAX, PyTorch, TensorFlow, NumPy. Familiarity with machine learning and RL, plus the mathematics and statistics knowledge needed to follow relevant research papers (linear algebra, calculus, etc). We're also interested More ❯

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 ❯

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 ❯

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 models More ❯

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 with More ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 potential … 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 health More ❯

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 potential … 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 health More ❯

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 models More ❯