trlx/docs/requirements.txt

accelerate==0.12.0
datasets==2.4.0
deepspeed==0.7.3
einops==0.4.1
numpy==1.23.2
sphinx==4.0.0
sphinx_rtd_theme
torchtyping
tqdm==4.64.0
transformers==4.21.2
wandb==0.13.2