2 of 2 Permanent JAX Jobs in Central London

applied AI research. Commercial or research experience implementing machine learning algorithms. Strong programming skills in Python; C++ experience is advantageous. Experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks. Why Apply? Join a world-class AI research team working on cutting-edge machine learning technologies. Play a key role ...

Responsibilities Advance applied research across the LLM stack, including model training, adaptation, inference, and agentic systems. Design, build, and release foundation-model and agentic systems that support high-value business workflows from prototype to production. ...