torch>=2.2
ocptv==0.1.7

[:sys_platform == "linux"]
nvidia-ml-py>=12.535

[test]
pytest>=8
numpy
jsonschema
