Metadata-Version: 2.4
Name: jaxviz
Version: 0.1.4
Summary: Interactive JAX forward-pass visualizer for notebooks
License: GPL-3.0-only
Keywords: jax,visualization,neural-network,sharding,spmd
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
Classifier: Natural Language :: English
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.9
License-File: LICENSE
License-File: NOTICE.txt
Requires-Dist: jax>=0.4.0
Requires-Dist: ipython>=7.0.0
Provides-Extra: test
Requires-Dist: pytest>=7.0; extra == "test"
Requires-Dist: flax>=0.8.0; extra == "test"
Dynamic: classifier
Dynamic: keywords
Dynamic: license
Dynamic: license-file
Dynamic: provides-extra
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary
