1 to 25 of 72 JAX Jobs in England

Data Scientist, AWS Generative AI Innovation Center Job ID: Amazon Web Services EMEA Dubai FZ Branch - Q29 Amazon launched the Generative AI Innovation Center (GenAIIC) in June 2023 to help AWS customers accelerate the use of generative AI to solve 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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 modern 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 ❯

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 ❯

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 ❯

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 ❯

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 ❯

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 ❯