code: def extract_kwargs(arg):
      kwargs = {
          "offsets": arg.offsets(),
          "_metadata_cache": arg._metadata_cache,
      }
      return kwargs
  
imports: import functools
  import math
  import operator
  import torch
  from torch.nested._internal.sdpa import jagged_scaled_dot_product_attention
  from .nested_tensor import NestedTensor
  from typing import *
  from torch.fx.operator_schemas import normalize_function
label_desc: Use image node_id 10 for calling a global function with example usage: extract_kwargs(arg) and returns: kwargs