├── .github └── workflows │ └── python-publish.yml ├── .gitignore ├── LICENSE ├── README.md ├── demo └── 1.gif ├── examples └── main.py ├── pyproject.toml └── src └── jax_smi ├── __init__.py ├── cli_tool.py └── initialise_tracking.py /.github/workflows/python-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/.github/workflows/python-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | build 2 | dist 3 | *.egg-info 4 | .mypy_cache 5 | /venv 6 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/README.md -------------------------------------------------------------------------------- /demo/1.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/demo/1.gif -------------------------------------------------------------------------------- /examples/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/examples/main.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/pyproject.toml -------------------------------------------------------------------------------- /src/jax_smi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/src/jax_smi/__init__.py -------------------------------------------------------------------------------- /src/jax_smi/cli_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/src/jax_smi/cli_tool.py -------------------------------------------------------------------------------- /src/jax_smi/initialise_tracking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ayaka14732/jax-smi/HEAD/src/jax_smi/initialise_tracking.py --------------------------------------------------------------------------------