It does sound like it could be overheating except for the fact that you mentioned it sometimes dies on takeoff. Try running a full tank of fuel through the engine on the ground while you vary the throttle to simulate flying. Check for any air bubbles going through the fuel lines, if there is, you have a leak somewhere. If it doesn't die during that test then I suspect the needles aren't set correctly, probably too lean but it's hard to say.