Can AI understand the theory of relativity?