Yes, the MB Air 2017 SSD (4 Lane PCIe 2.0 x4 flash storage) is the same as from 2015 model.
And you can upgrade all of them to a Samsung EVO M.2 SSD too, model lines 960 and 970 (Both are PCIe 3.0 x4 but 970 has NVMe and is faster, macOS High Sierra required for NVMe compatibility). Both 960 and 970 are faster than the original Apple SSD, and cheaper too.
The OWC Aura Pro X is PCIe 3.0 x4 NVMe SSD too, hope as good as Samsung ones.
The "Original Apple SSD" is manufactured by Samsung, but the connector pin-out is different of M.2 standard. Fortunately there are compatibility adapters to install M.2 SSD into Macs (like this: https://www.globalmac.cl/Adaptador-Corto...)
Please use Carbon Copy Cloner to clone your SSD (already upgraded to macOS Sierra or High Sierra) to an external disk. Then replace the SSD with proper adapter. Boot from external disk and use CCC to clone all your data to the new SSD. (This procedure will not restore BootCamp Partition, you will need WinClone for that).
Go on with your upgrade, this is cool!