With the new driver, we should be able to export more metrics, e.g. memory usage.
Check what numbers we can shake out of rocm-smi and include the useful ones in the Python script for export.
If there is a lot of additional metrics, consider refactoring the parsing/metric-assignment part of the code.