It won't be an easy swap to anything else.
Everything is slightly different to a comparable year sti.
An 01 will need a speed signal from the trans which later ones take from the ABS system instead so you need to install the sensor for that also.
Those rear axles are weird ones with r180 size on the inside and r160 on the outside