Could be a couple of things:
Iv seen a fuel pump do this but since only at the lights it shuts off it's probably not that.
Try cleaning out your intake etc.. (Attachments)
Does the Engine does immeditatly? or does it stutter off first?

Try out the relay make sure it works.

 

it does it sometimes and sometimes it doesn't, it seem to be related to temperature, in the morning when i get up to go to work (5 A.M.) it will shut off at the first stop light but after that it is fine for the rest of the day, when i get off work around 4P.M. it won't do it until i have riden for a while and then it wil studder and jerk, but when i get to a stop it will not do it it will simiply shut off,

 

You're problem is going to be the throttle position sensor. Had the same problem with my old 99 Club Cab with 5.2L and my 01 Dakota Quad Cab with the 5.9L.

 

still loking for some help with my problem, got a engine code of 601 and i am about to give up,it will shut down at a stop light and when i am driving over 55 mph it is acting like it is surging or wants to shut down but i give it mor gas and it keeps running, I NEED HELP, changed the IAC, TPS,and cleaned out the throttle body and still have the same problem.....I an losing hair over this....