Mycelium • Graph visualization library

Visualize your machine learning model

Mycelium is a library for creating graph visualizations of machine learning models or any other directed acyclic graphs. It also powers the graph viewer of the Talaria model visualization and optimization system.

Batteries included

Everything you need to visualize the structure of your machine learning models

Mycelium makes it easy to visualize exactly the characteristics of your machine learning models that your interested in.

The following authors contributed directly to Mycelium.

Acknowledgements

This project is a collaboration across multiple teams at Apple. We would like to extend our thanks to all authors of Talaria: Chaoqun Wang, Jinmook Lee, Dominik Moritz, Jeffrey Bigham, Zhile Ren, Cecile Foret, and Qi Shan.

We would love to hear from you!

Attribution

You can use to following BibTex entry to cite Talaria (and therefore Mycelium).

@inproceedings{Hoh+2024,
  title={Talaria: Interactively Optimizing Machine Learning Models for Efficient Inference},
  author={Fred Hohman and Chaoqun Wang and Jinmook Lee and Jochen Görtler and Dominik Moritz and Jeffrey Bigham and Zhile Ren and Cecile Foret and Qi Shan and Xiaoyi Zhang},
  booktitle={Proceedings of the SIGCHI Conference on Human Factors in Computing Systems},
  year={2024},
  organization={ACM},
  doi={10.1145/3613904.3642628}
  url = {https://arxiv.org/abs/2404.03085}
}