code: def matmul(X, W):
      return explicit_shard_W_axis_1(X, W)
  
imports: import numpy
  from tinygrad import Tensor, Device, GlobalCounters
  from tinygrad.helpers import Timing
label_desc: Use image node_id 2 for calling a global function with example usage: matmul(X, W) and returns: explicit_shard_W_axis_1