26 to 33 of 33 JAX Jobs in the UK

Developer will work with a variety of technologies including the Python scientific stack (NumPy, Pandas, SciPy, etc.), frameworks such as OpenCV, PyTorch, TensorFlow and JAX, data pipeline tools such as Airflow or Prefect and data processing tools such as Spark or Databricks. The Company The Python Developer will be joining ...

machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down ...

Mathematics, Physics, Computer Science, or a related field Strong hands-on experience building, training, and scaling foundation models with modern ML tooling (PyTorch, Ray, JAX) Strong background in graph neural networks and transformer architectures Demonstrated ability to turn research ideas into robust, scalable implementations Comfortable owning whatever needs doing, from ...

combined with hands-on experience with LLM post-training, alignment, or reinforcement learning Proficiency in Python and at least one major DL framework (PyTorch, JAX, or TensorFlow) Experience training large models on distributed systems Publications at top-tier conferences (NeurIPS, ICML, ICLR, ACL, CVPR, etc.) Comfortable working in fast-moving ...

. Domain Depth: Hands-on expertise in VLMs , Diffusion/Generative Video , or LLM Pre-training . Frameworks: Mastery of Python and PyTorch or JAX . Systems: Experience with distributed training and complex numerical debugging. Randstad Technologies Ltd is a leading specialist recruitment business for the IT & Engineering industries. Please ...

production systems Hands-on post-training experience (RLHF, DPO, or similar) and deployed LLMs in production Strong Python and modern ML tooling (PyTorch/JAX, training infrastructure, evaluation pipelines) Track record shipping research from prototype to production impact Comfortable owning work end-to-end: architecture to data pipelines to production ...

depending on experience. Experience and skills required for the Agentic AI Engineer, Agentic AI Developer, Python Fluency with modern ML toolchains (PyTorch/JAX, RL or behaviour-cloning libraries, agentic frameworks, MCP, distributed training workflows) click apply for full job details ...

with teleops teams to transform human trajectories into robust robotic skills through behavior cloning. High-Performance Engineering: Designing and profiling research-grade PyTorch/JAX code to support large-scale, distributed RL infrastructure. Essential Skills Needed Deep Learning Mastery: 5+ years building and shipping models, with deep hands-on expertise … Expert: A proven track record of solving complex, real-world problems using Deep Reinforcement Learning. Technical Rigour: Mastery of Python and PyTorch/JAX, including the ability to profile performance and debug complex numerical stability issues. Robotics Foundation: Practical experience with simulators (Isaac Sim/MuJoCo) and a deep understanding ...