sub\_modules ============ .. automodule:: pytorch_forecasting.models.nhits.sub_modules .. rubric:: Functions .. autosummary:: :toctree: :template: custom-base-template.rst init_weights .. rubric:: Classes .. autosummary:: :toctree: :template: custom-class-template.rst IdentityBasis NHiTS NHiTSBlock StaticFeaturesEncoder