Yes, you can freeze babka at pretty much any stage of the recipe, from the brioche dough, to the shaped and unbaked babka, or the baked babka. I prefer to freeze baked babka because yeast loses some activity in the freezer. Double wrap the babka in plastic wrap and seal it in a freezer bag. You can let it come to room temperature or gently warm it wrapped in foil in the oven.