numpy
torch
fire
sentencepiece
tiktoken
transformers
ipykernel
ipywidgets
matplotlib
networkx
black
flake8
flake8-pyproject
isort
