How AI can help reveal the workings of the brain