The US ambassador to NATO has said the military alliance is trying to expedite the process to ensure that Sweden and Finland join in the next "couple of months".