VizNeuron

VizTracer plugins for AWS Neuron. It is inspired by the official VizTracer plugings: vizplugins.

Features

VizNeuron currently only supports memory usage tracing via Neuron Sysfs Filesystem.

Install

The preferred way to install VizNeuron is via pip:

pip install vizneuron

Basic Usage

VizNeuron should be used with VizTracer.

You can use VizNeuron and the plugin via command line:

viztracer --plugin vizneuron.memory_usage -- my_script.py arg1 arg2

Or equivalent syntax:

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.

Finally, VizNeuron provides a PyTorch Lightning Callback that can be used as part of model training.