There are a couple of international schools in Lausanne, where your son would be able to learn in English. Have a look at our page on education in Switzerland for some info on the school system.
Keep in mind that international schools often have long waiting lists, so I advise you to get onto it as soon as possible. It can also be quite expensive. Does your partner have an education allowance included in their package? Will you be working in Switzerland?