$ bash run_llama31.sh
Traceback (most recent call last):
  File "/workspace/llama31/pretrain_llama31.py", line 18, in <module>
    from nemo.collections import llm
  File "/opt/NeMo/nemo/collections/llm/__init__.py", line 42, in <module>
    from nemo.collections.llm.gpt.data.api import dolly, hf_dataset, mock, squad
  File "/opt/NeMo/nemo/collections/llm/gpt/data/api.py", line 16, in <module>
    import nemo_run as run
  File "/root/.local/lib/python3.10/site-packages/nemo_run/__init__.py", line 28, in <module>
    from nemo_run.help import help
  File "/root/.local/lib/python3.10/site-packages/nemo_run/help.py", line 30, in <module>
    from nemo_run.cli.api import _load_entrypoints, _load_workspace
  File "/root/.local/lib/python3.10/site-packages/nemo_run/cli/__init__.py", line 16, in <module>
    from nemo_run.cli.api import (
  File "/root/.local/lib/python3.10/site-packages/nemo_run/cli/api.py", line 70, in <module>
    from nemo_run.cli.config import ConfigSerializer
  File "/root/.local/lib/python3.10/site-packages/nemo_run/cli/config.py", line 18, in <module>
    import toml
ModuleNotFoundError: No module named 'toml'