requirements.txt

################################
###### Core dependencies #######
################################
cached-path>=1.0,<2.0
jsonnet>=0.10 ; sys.platform != 'win32'
GitPython>=3.0,<4.0
PyYAML>=5.4.1,<7.0
dill
base58
xxhash
filelock>=3.4,<4.0
click>=8.0,<9.0
click-help-colors>=0.9.1,<0.10
rich>=12.3,<13.0
tqdm>=4.62,<5.0
more-itertools>=8.0,<10.0
sqlitedict
glob2>=0.7
petname>=2.6,<3.0
pytz

##################################################
###### Extra dependencies for integrations #######
##################################################
# NOTE: we use a special trailing comment on each line to denote which extras
# each package is needed by. For example, PyTorch is needed by the 'torch' extra
# that you install with 'pip install tango[torch]'.
torch>=1.9,<1.14             # needed by: torch,transformers,fairscale
numpy                        # needed by: torch,transformers,fairscale
datasets>=1.12,<3.0          # needed by: flax,datasets,transformers
wandb>=0.12,<0.14            # needed by: wandb
retry                        # needed by: wandb
transformers>=4.12.3         # needed by: transformers
sentencepiece==0.1.97        # needed by: transformers
fairscale>=0.4.6,<0.5        # needed by: fairscale
beaker-py>=1.11.6,<2.0       # needed by: beaker

# sacremoses should be a dependency of transformers, but it is missing, so we add it manually.
sacremoses                   # needed by: transformers
jax>=0.3.13                  # needed by: flax
flax>=0.5.0                  # needed by: flax
optax>=0.1.2                 # needed by: flax
tensorflow-cpu>=2.9.1        # needed by: flax