US looking at restricting Chinese investment in artificial intelligence on security grounds