Add a few more comments for inference modules