could be pressure switch issue, not bad enough to throw a code but enough to create issues when driving.
its possible some of the electronic components on the old transmission, were transferred over to the new transmission.
which is why your problem has returned.
could also be temperature sensor related, are you noticing it more when the vehicle is cold and less when warmed up?