How special relativity can help AI predict the future