Update jit graph visualization

[xiguadong]

No description provided.

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[codecov]

Codecov Report

Merging #126 (8c10024) into master (cbe5cf0) will not change coverage.
The diff coverage is n/a.

❗ Current head 8c10024 differs from pull request most recent head c7839b2. Consider uploading reports for the commit c7839b2 to get more accurate results

@@           Coverage Diff           @@
##           master     #126   +/-   ##
=======================================
  Coverage   83.54%   83.54%           
=======================================
  Files           8        8           
  Lines         784      784           
=======================================
  Hits          655      655           
  Misses        129      129           

Flag	Coverage Δ
unittests	83.54% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update cbe5cf0...c7839b2. Read the comment docs.

[zhiqwang]

LGTM! Thanks @xiguadong for your contribution, and I just cleanup some scripts based on your implementation.