LICENSE
MANIFEST.in
NOTICE.txt
README.md
pyproject.toml
requirements.txt
setup.py
jaxviz/__init__.py
jaxviz/enums.py
jaxviz/graph_transforms.py
jaxviz/hlo_to_graph.py
jaxviz/jaxpr_to_graph.py
jaxviz/render.py
jaxviz.egg-info/PKG-INFO
jaxviz.egg-info/SOURCES.txt
jaxviz.egg-info/dependency_links.txt
jaxviz.egg-info/requires.txt
jaxviz.egg-info/top_level.txt
jaxviz/adapters/__init__.py
jaxviz/adapters/nnx.py
jaxviz/assets/__init__.py
jaxviz/assets/d3.min.js
jaxviz/assets/jsoneditor-10.2.0.min.css
jaxviz/assets/jsoneditor-10.2.0.min.js
jaxviz/assets/viz-standalone.js
jaxviz/templates/__init__.py
jaxviz/templates/example.html
jaxviz/templates/graph.html
tests/test_build_graph.py
tests/test_examples_generator.py
tests/test_hlo_shapes.py
tests/test_trace_model.py