Hi!
The explanation of forward_op is a bit confusing and I think that it has some typos.

It is not clear what this means: Given a dictionary input x, it will return a dictionary with the same keys and the values applied by the corresponding values of the module_dict with specified parameters.
"The keys in x are same with the keys in the module_dict." I think that this may mean the keys in x and the module_dict are the same. Or, the keys in x are equal to the keys in the module_dict.
Hi!

The explanation of
forward_opis a bit confusing and I think that it has some typos.It is not clear what this means: Given a dictionary input x, it will return a dictionary with the same keys and the values applied by the corresponding values of the module_dict with specified parameters.
"The keys in x are same with the keys in the module_dict." I think that this may mean the keys in x and the module_dict are the same. Or, the keys in x are equal to the keys in the module_dict.