diff --git a/pyproject.toml b/pyproject.toml index ca440a0..9612019 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -14,11 +14,9 @@ packages = [{ include = "qiboml", from = "src" }] python = ">=3.9,<3.12" numpy = "^1.26.4" numba = "^0.59.0" -tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux' or sys_platform == 'darwin'" } +tensorflow = "^2.16.1" torch = "^2.2.0" jax = "^0.4.25" -# TODO: the marker is a temporary solution due to the lack of the tensorflow-io 0.32.0's wheels for Windows, this package is one of -# the tensorflow requirements qibo = "^0.2.6" [tool.poetry.group.dev]