code: def compose(*fns):
      def compose2(f, g):
          return lambda *a, **kw: f(g(*a, **kw))
  
      return reduce(compose2, fns)
  
imports: import itertools
  from functools import partial, reduce
  from typing import Iterator
  import timm
  import torch
  import torch.nn
  from timm.models.layers import Mlp
  from timm.models.vision_transformer import Attention
  from timm.models.vision_transformer import Block
  from torch.utils import benchmark
  from utils import benchmark_main_helper
  import xformers.ops
label_desc: Use image node_id 4 for calling a global function with example usage: compose() and returns: reduce, function