Thrust bearing assembly