torch>=2.0.0
numpy
transformers
datasets
tiktoken
wandb
tqdm
PyYAML
lightning
tensorboard
fastapi
uvicorn
gymnasium

[dev]
black==23.3.0
pytest
pytest-cov
pytest-sugar
build
twine
httpx
