code: def init_weights(module: nn.Module, scale: float = 1.0):
      if type(module) == nn.Linear:
          nn.init.uniform_(module.weight, -scale, scale)
          nn.init.zeros_(module.bias)
  
imports: from functools import partial
  from typing import Optional, Callable, List, Union
  import torch
  from torch import nn
  from torch.distributions import Categorical, MixtureSameFamily, Normal
  from gluonts.core.component import validated
  from gluonts.torch.distributions import DiscreteDistribution
  from .scaling import min_max_scaling, standard_normal_scaling
label_desc: Use image node_id 1 for calling a global function with example usage: init_weights(module, scale) without return types