torch>=2.2.0
click>=8.0.0
pyyaml>=6.0
numpy>=1.24.0
tabulate>=0.9.0
transformers>=4.36.0
datasets>=2.16.0

[all]
transformer-engine>=2.7.0
jax>=0.4.30
jaxlib

[jax]
jax>=0.4.30
jaxlib

[te]
transformer-engine>=2.7.0
