It depends on where you want to make better acceleration. A lighter flywheel will make low speed acceleration a lot better, whereas lighter wheels/tyres and drive shaft will make high speed acceleration better. The difference being the gearing.
My guess is that there is more to be gained in tyre selection than in the driveshaft.