Sounds like the torque converter.

For a simple test, get up to about 50-60 MPH (enough to have it in 3rd or 4th gear with the TC locked). Then punch the gas quickly, but not hard, just enough to unlock the TC but not drop down a gear. If you get jerkiness or shuddering, it's a bad unit.