https://github.com/clumsy/vizneuron
VizTracer plugins for AWS Neuron
https://github.com/clumsy/vizneuron
Last synced: 3 months ago
JSON representation
VizTracer plugins for AWS Neuron
- Host: GitHub
- URL: https://github.com/clumsy/vizneuron
- Owner: clumsy
- License: apache-2.0
- Created: 2024-09-20T14:58:41.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2024-09-20T22:18:00.000Z (over 1 year ago)
- Last Synced: 2025-09-17T04:16:07.059Z (9 months ago)
- Language: Python
- Size: 8.79 KB
- Stars: 2
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# VizNeuron
[VizTracer](https://github.com/gaogaotiantian/viztracer) plugins for AWS Neuron.
It is inspired by the official VizTracer plugings: [vizplugins](https://github.com/gaogaotiantian/vizplugins).
# Features
VizNeuron currently only supports memory usage tracing via [Neuron Sysfs Filesystem](https://awsdocs-neuron.readthedocs-hosted.com/en/latest/tools/neuron-sys-tools/neuron-sysfs-user-guide.html#neuron-sysfs-filesystem-structure).
# Install
The preferred way to install VizNeuron is via pip:
```sh
pip install vizneuron
```
# Basic Usage
VizNeuron should be used with VizTracer.
You can use VizNeuron and the plugin via command line:
```sh
viztracer --plugin vizneuron.memory_usage -- my_script.py arg1 arg2
```
Or equivalent syntax:
```sh
viztracer --plugin "vizneuron --memory_usage" -- my_script.py arg1 arg2
```
You can also add the plugin to VizTracer programmatically as described in the [official documentation](https://viztracer.readthedocs.io/en/latest/plugins.html).
Finally, VizNeuron provides a [PyTorch Lightning Callback](https://lightning.ai/docs/pytorch/stable/extensions/callbacks.html) that can be used as part of model training.