code: def collect_parameters(self):
      model_parameters = {
          n: p
          for n, p in self.named_parameters()
          if not n.endswith(".quantiles") and p.requires_grad
      }
      return model_parameters, None
  
imports: from typing import Callable, NamedTuple, Optional
  import torch
  import torch.nn
  from torch import Tensor
  import neuralcompression.functional
label_desc: Use image node_id 2 for calling the DvcStage1 obj's underlying member method code with example usage: obj.collect_parameters() and returns: model_parameters,  None