# NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package
#  in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment

# colossalai>=0.1.10  # TODO: uncomment when there's a stable version released
fairscale>=0.4.5, <0.4.13
deepspeed>=0.6.0, <=0.7.0
# no need to install with [pytorch] as pytorch is already installed
horovod>=0.21.2, !=0.24.0, <=0.26.1
hivemind==1.1.2; sys_platform == 'linux'
