Will a 4l60e transmission from a 98 k1500 4wd 5.7l work in a 96 k1500 4wd 5.7l?

Yes, it's the same transmission. Direct swap should do it. Now, if you were trying to swap in a 4L80, then it would get a bit more complicated. But two 4L60Es will swap out evenly.