151 to 175 of 290 JAX Jobs in England

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks #J-18808-Ljbffr More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse Preferred: Experience in LLMs, reinforcement learning, or chip design Familiarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid work environment Opportunities to publish More ❯

You’ll need to be both a scientist and an engineer: excited by innovation, and pragmatic about delivering impact! 🛠️ Key Technologies: Python and modern ML libraries (e.g. PyTorch, TensorFlow, JAX) AWS and cloud infrastructure for scalable training and inference ML ops tooling, model monitoring, and data pipelines Bonus: experience with LLMs & generative AI is a strong plus 🎯 What They’re More ❯

You’ll need to be both a scientist and an engineer: excited by innovation, and pragmatic about delivering impact! 🛠️ Key Technologies: Python and modern ML libraries (e.g. PyTorch, TensorFlow, JAX) AWS and cloud infrastructure for scalable training and inference ML ops tooling, model monitoring, and data pipelines Bonus: experience with LLMs & generative AI is a strong plus 🎯 What They’re More ❯

You’ll need to be both a scientist and an engineer: excited by innovation, and pragmatic about delivering impact! 🛠️ Key Technologies: Python and modern ML libraries (e.g. PyTorch, TensorFlow, JAX) AWS and cloud infrastructure for scalable training and inference ML ops tooling, model monitoring, and data pipelines Bonus: experience with LLMs & generative AI is a strong plus 🎯 What They’re More ❯

You’ll need to be both a scientist and an engineer: excited by innovation, and pragmatic about delivering impact! 🛠️ Key Technologies: Python and modern ML libraries (e.g. PyTorch, TensorFlow, JAX) AWS and cloud infrastructure for scalable training and inference ML ops tooling, model monitoring, and data pipelines Bonus: experience with LLMs & generative AI is a strong plus 🎯 What They’re More ❯

You’ll need to be both a scientist and an engineer: excited by innovation, and pragmatic about delivering impact! 🛠️ Key Technologies: Python and modern ML libraries (e.g. PyTorch, TensorFlow, JAX) AWS and cloud infrastructure for scalable training and inference ML ops tooling, model monitoring, and data pipelines Bonus: experience with LLMs & generative AI is a strong plus 🎯 What They’re More ❯

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

of experience in building production quality ML systems. Preferred Qualifications: Master's degree or PhD in Computer Science or a related technical field. Experience with modern ML frameworks (e.g., JAX, Pytorch or TensorFlow). Experience with launching applied Machine Learning/Natural Language Processing (ML/NLP) projects. About the Job Google's software engineers develop the next-generation technologies More ❯

autonomy and a collaborative mindset. Expertise in machine learning applied to proteins, small molecules, and molecular simulations. Strong programming skills in Python and proficiency with ML frameworks (e.g., PyTorch, JAX, TensorFlow). Hands-on experience with scientific computing libraries (e.g., RDKit, OpenMM) and molecular representations (e.g., SMILES, molecular graphs, 3D structures). Ability to bridge AI research with practical drug More ❯

on distributed training infrastructures. Familiarity with evaluating and improving the generalisability and robustness of ML systems. Proficiency in programming languages such as Python and ML frameworks (e.g., PyTorch, TensorFlow, JAX). Excellent communication skills to collaborate effectively with cross-functional teams and present findings. One or more papers at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR More ❯

such as Python or C++). Knowledge of linear algebra, calculus and statistics equivalent to at least first-year university coursework. Experience exploring, analysing and visualising data. Experience using Jax, PyTorch, NumPy, Pandas or similar ML/scientific libraries. In addition, the following would be an advantage: Experience with advanced recent LLM techniques. Scientific domain knowledge (particularly biology, physics, or More ❯

systems, resource-efficient AI data and algorithm scaling, or neural network architectures. Experience communicating complex research for public audiences of peers. Experience with deep learning frameworks such as Pytorch, Jax, or Tensorflow. Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment. Preferred Qualifications Proven track record of achieving significant More ❯