lightning >=2.0.0
# note: when you bump the latest version, update also docker image for GPU testing
colossalai >=0.2.0
torch <2.0.0  # ModuleNotFoundError: No module named 'torch._six'
