code: def test_jit_trace(model):
      zeros_input = model.describe_inputs().zeros()
      ones_input = valmap(torch.ones_like, zeros_input)
  
      script_module = torch.jit.trace(
          model, tuple(zeros_input.values())
      )
  
      torch.manual_seed(0)
      output_1 = model(**ones_input)
  
      torch.manual_seed(0)
      output_2 = script_module(**ones_input)
  
      assert all_equal(output_1, output_2)
  
imports: import pytest
  from toolz import valmap
  import torch
  from gluonts.torch.model.deepar import DeepARModel
  from gluonts.torch.model.tft import TemporalFusionTransformerModel
  from gluonts.torch.model.simple_feedforward import SimpleFeedForwardModel
label_desc: Use image node_id 2 for calling a global function with example usage: test_jit_trace(model) without return types