numpy
torch
packaging
tqdm
pyyaml
transformers
datasets
protobuf
SentencePiece
rich
omegaconf
pydantic
click
click_pathlib
jq
class_resolver
matplotlib
wandb
einops>=0.7.0
debugpy

[install_helper]
ninja

[linting]
pre-commit

[tests]
pytest
pytest-cov
debugpy
