26 to 50 of 117 JAX Jobs in London

GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g., TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ More ❯

GCP) AI/ML services, especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). • Proficiency in Python and strong experience with AI/ML frameworks (e.g., TensorFlow, PyTorch, JAX, Hugging Face Transformers). • Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). • Solid understanding of core AI/ More ❯

models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with Deep Learning techniques (e.g. Network architectures, regularization techniques, learning techniques, loss-functions, optimization strategies, etc.) Familiarity with PyTorch , TensorFlow , JAX or similar frameworks Strong coding abilities in Python and/or C++ Preferred Qualifications: 2D & 3D Generative AI Reinforcement Learning LLMs and Natural Language Processing Computational geometry and geometric methods 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 ❯

models, dimension reduction, clustering, classification, regression techniques, etc.) Familiarity with Deep Learning techniques (e.g. Network architectures, regularization techniques, learning techniques, loss-functions, optimization strategies, etc.) Familiarity with PyTorch , TensorFlow , JAX or similar frameworks Strong coding abilities in Python and/or C++ Preferred Qualifications 2D & 3D Generative AI Reinforcement Learning LLMs and Natural Language Processing Computational geometry and geometric methods 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 ❯

years of experience in a data-driven role, with exposure to: building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; developing models for bespoke problem settings that involve high-dimensional data (spatiotemporal, geometric, physical); iterating on network architectures and model structure, tuning and optimising for More ❯

analysis, and visualisation Strong knowledge of linear algebra, calculus, probability, and statistics Demonstrated ability to write clean, idiomatic, and highly performant Python code Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas Expertise with detailed data quality control procedures and data visualisation Experience with experimental design and statistical analysis Demonstrated 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 ❯

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 ❯

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 ❯

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 ❯

custom in-house). Solid grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis. Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX). Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers. Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one More ❯

languages such as Python, C++ or Rust - Hands-on experience with parallel computing platforms such as CUDA, OpenMP, etc - Deep understanding of AI frameworks such as PyTorch, TensorFlow, and JAX, and their demands on underlying compute infrastructure, memory bandwidth, network interconnect, and storage as scale goes up - Knowledge of emerging AI hardware accelerators and architectures - Experience with containerization and orchestration More ❯

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, and More ❯