Enabling AI for High Energy Physics in the UK