RLHF Module
===========

.. autoclass:: chatlearn.models.base_module.BaseModule
    :members:
    :undoc-members:

.. autoclass:: chatlearn.models.torch_module.TorchModule
   :members:
   :undoc-members:
   :show-inheritance:

.. autoclass:: chatlearn.models.megatron_module.MegatronModule
   :members:
   :undoc-members:
   :show-inheritance: