Alternatively you could try to use one of the programs listed in this thread 133.
However, I am not sure whether any of these options are runnable in python. If not you might be able to use them together with torch.jit as it is mentioned in Road to 1.0 18