Yes, non-EU can choose the country in which it will get registered for OSS, the only condition is that the NON-EU company needs to have a standard VAT registration in that country.