I would check all small stuff first before you assume its in the trans!
Check all drive pins in the axles & drive shafts including the drive cups that
are attached to the diffs with grub screws ,those are proned to work loose or fall out
an give you false failures on other stuff!
So always...