smdebug/mxnet/hook.py
smdebug/pytorch/hook.py

smdebug/mxnet/collection.py
smdebug/pytorch/collection.py