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.
- Performance
- Based on SVG but can handle models with thousands of nodes through hierarchical clustering.
- Model Support
- Comes with ONNX support out of the box. By writing a loader you can use Mycelium with any graph or model format.
- Embeddable
- The graph viewer—although written in Svelte—is framework agnostic and can be embedded into any application by mounting it directly to the DOM.
- Customizable
- Mycelium leverages SVG together with a tiny UI framework to allow rich contents in nodes and tooltips.
Team
Authors
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}
}