Created by: yukavio
Add an argument for the user can get Program as return value which is easier for user to load a quantized model checkpoint.