51 to 74 of 74 Google JAX Jobs in the UK excluding London

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis More ❯

knowledge by documenting systems, debugging approaches, and lessons learned You May Be a Good Fit If You: Have hands-on experience training large language models, or deep expertise with JAX, TPU, PyTorch, or large-scale distributed systems Genuinely enjoy both research and engineering work-you'd describe your ideal split as roughly 50/50 rather than heavily weighted … craft as a research engineer Care about the societal impacts of AI and responsible scaling Strong Candidates May Also Have: Previous experience training LLM's or working extensively with JAX/TPU, PyTorch, or other ML frameworks at scale Contributed to open-source LLM frameworks (e.g., open_lm, llm-foundry, mesh-transformer-jax) Published research on model training More ❯

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 to integrate seamlessly into daily working life. We democratize AI through high-performance, optimized More ❯

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning Consultant, you will play a More ❯

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning Consultant, you will play a More ❯