lightning>=2.0.0
bagua-cuda116>=0.9.2

[test]
codecov>=2.1
coverage>=5.0
mypy==1.0.1
pytest-cov
pytest>=6.0
