class_docstr: Helper class to create graphs to display memory
class_docstr_tok: ["Helper","class","to","create","graphs","to","display","memory"]
code: def _y_axis_in_gigabytes(ax):
      ax.ticklabel_format(axis="y", style="sci", scilimits=(9, 9))
  
imports: from contextlib import contextmanager
  from dataclasses import dataclass
  from enum import auto, Enum
  from functools import lru_cache
  from typing import Dict, List, NamedTuple, Tuple, Union
  import numpy
  import torch
  import torch.nn
  from fairscale.nn import FullyShardedDataParallel
  from vissl.utils.visualize import matplotlib_figure_to_image
label_desc: Use image node_id 8 for calling the MemoryGraphCreator obj's underlying member method code with example usage: obj._y_axis_in_gigabytes(ax) without return types