code: def save_graph(symbol: mx.sym.Symbol, filename: str, hide_weights: bool = True):
      
      dot = mx.viz.plot_network(symbol, hide_weights=hide_weights)
      dot.render(filename=filename)
docs: Dumps computation graph visualization to .pdf and .dot file.
  
      :param symbol: The symbol representing the computation graph.
      :param filename: The filename to save the graphic to.
      :param hide_weights: If true the weights will not be shown.