AI Research Engineer

In this role you will advance brain-inspired, mathematically grounded AI methods to improve the efficiency of future models. You will work across areas such as brain-inspired architectures, neural memory, optimization, and hardware-aware ML, collaborating with a cross-disciplinary team. You will design and run rigorous experiments, develop implementations in Python and PyTorch (or JAX), and contribute to research outputs and open-source projects. You help shape scalable, energy-conscious AI systems with real-world impact.

• Develop and evaluate brain-inspired architectures and memory models
• Design and run experiments with baselines, ablations, robustness, and computational cost analysis
• Implement and iterate in Python and PyTorch (or JAX) with awareness of transformers and LLMs
• Collaborate with research teams to advance methods and models
• Contribute to publications or open-source projects
• Analyze training dynamics and efficiency to drive improvements
• Explore hardware-aware machine learning aspects like memory bandwidth and energy modelling

• Bachelor (or higher) in a quantitative field
• Strong foundations in linear algebra, probability, optimisation, dynamical systems, numerical methods, information theory or statistical learning
• Proficiency in Python and PyTorch (or JAX) with solid understanding of transformers/LLMs
• Rigorous experimental mindset bridging theory, experimentation and implementation
• Research publications, open-source contributions or substantial independent projects
• Location: London, full-time, permanent

• collaboration
• clear communication
• curiosity
• Python
• PyTorch
• JAX