If you want to avoid reassigning hardware every time you use MainStage, especially when traveling or using multiple different keyboards, you can create layout files in MainStage.
Watch this tutorial to learn how:
Check out the Sunday Sounds blog for many more great MainStage and Ableton Live tutorials.