Check your ignition system etc. Tune up with plugs and wires. Ensure the coils are all good. I've seen a similar issue with a coil pack and after replacing a trans.. yeah.. still had to shotgun parts to find it. I'd shotgun another set of coils/icm before you swap a trans.