torch
transformers
nemo_toolkit['all']
numpy