51 to 75 of 75 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance 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, instance More ❯

and trading concepts. Nice to Have Experience with crypto exchanges and market microstructure. Hands-on with Bokeh or other interactive viz libraries. Distributed compute (Dask, Ray) experience. ML Stack (JAX, Pytorch, Tensorflow, XGBoost, etc.) experience. Experience with compilers and code generation. Benefits International environment (English is the main language) Pension 100% health coverage Team events and offsites Recruitment process Are 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 toward … 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, scaling laws 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 ❯

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 ❯