code: def write_jsonl(filename, data):
      if not os.path.exists(os.path.dirname(filename)):
          os.makedirs(os.path.dirname(filename))
  
      with open(filename, "wb") as f:
          for x in data:
              f.write((json.dumps(x) + "\n").encode("utf-8"))
  
imports: import argparse
  import json
  import os
  from pathlib import Path
  import random
  from time import time
  import torch
  from transformers import GPT2TokenizerFast
  from jaxformer.hf.codegen.modeling_codegen import CodeGenForCausalLM
label_desc: Use image node_id 4 for calling a global function with example usage: write_jsonl(filename, data) without return types