2 of 2 JAX Jobs in Cambridgeshire

steps clearly to a multidisciplinary team. Required Qualifications Strong practical experience implementing and training reinforcement learning systems using modern ML tooling (e.g., PyTorch/JAX and RL libraries or custom code). Proven ability to take ownership of R&D work: formulate hypotheses, prototype quickly, evaluate rigorously, and iterate toward ...

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 Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid ...