Mark Zuckerberg's controversial company is helping create an AI-enabled headset that will be used to train members of the armed forces.