Migrate ML backends to Qiboml

pyproject.toml

@@ -33,8 +33,6 @@ tabulate = "^0.9.0"
 openqasm3 = { version = ">=0.5.0", extras = ["parser"] }
 numpy = "^1.26.4"
 networkx = "^3.2.1"
-tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux' or sys_platform == 'darwin'", optional = true }
-torch = { version = "^2.1.1", optional = true }
 
 [tool.poetry.group.dev]
 optional = true
@@ -69,8 +67,7 @@ scikit-learn = "^1.2.1"
 pytest-cov = "^4.0.0"
 pylint = "3.1.0"
 matplotlib = "^3.7.0"
-tensorflow = { version = "^2.16.1", markers = "sys_platform == 'linux'" }
-torch = "^2.1.1"
+qiboml = { git  = "https://github.com/qiboteam/qiboml.git" }
 qibojit = { git = "https://github.com/qiboteam/qibojit.git" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 stim = "^1.12.0"
@@ -105,8 +102,6 @@ qibojit = { git = "https://github.com/qiboteam/qibojit.git" }
 qibotn = { git = "https://github.com/qiboteam/qibotn.git" }
 
 [tool.poetry.extras]
-tensorflow = ["tensorflow"]
-torch = ["torch"]
 qinfo = ["stim"]
 
 [tool.pylint.main]

src/qibo/backends/__init__.py

@@ -7,12 +7,19 @@
 from qibo.backends.clifford import CliffordBackend
 from qibo.backends.npmatrices import NumpyMatrices
 from qibo.backends.numpy import NumpyBackend
-from qibo.backends.pytorch import PyTorchBackend
-from qibo.backends.tensorflow import TensorflowBackend
 from qibo.config import log, raise_error
 
-QIBO_NATIVE_BACKENDS = ("numpy", "tensorflow", "pytorch")
-QIBO_NON_NATIVE_BACKENDS = ("qibojit", "qibolab", "qibo-cloud-backends", "qibotn")
+QIBO_NATIVE_BACKENDS = ("numpy",)
+QIBO_NON_NATIVE_BACKENDS = (
+    "qibojit",
+    "qibolab",
+    "qibo-cloud-backends",
+    "qibotn",
+)
+QIBOML_BACKENDS = (
+    "tensorflow",
+    "pytorch",
+)
 
 
 class MetaBackend:
@@ -31,10 +38,6 @@ def load(backend: str, **kwargs) -> Backend:
 
         if backend == "numpy":
             return NumpyBackend()
-        elif backend == "tensorflow":
-            return TensorflowBackend()
-        elif backend == "pytorch":
-            return PyTorchBackend()
         elif backend == "clifford":
             engine = kwargs.pop("platform", None)
             kwargs["engine"] = engine
@@ -194,7 +197,7 @@ def _check_backend(backend):
 def list_available_backends() -> dict:
     """Lists all the backends that are available."""
     available_backends = MetaBackend().list_available()
-    for backend in QIBO_NON_NATIVE_BACKENDS:
+    for backend in QIBO_NON_NATIVE_BACKENDS + ("qiboml",):
         try:
             module = import_module(backend.replace("-", "_"))
             available = getattr(module, "MetaBackend")().list_available()
@@ -218,6 +221,10 @@ def construct_backend(backend, **kwargs) -> Backend:
     elif backend in QIBO_NON_NATIVE_BACKENDS:
         module = import_module(backend.replace("-", "_"))
         return getattr(module, "MetaBackend").load(**kwargs)
+    elif backend in QIBOML_BACKENDS:
+        module = import_module("qiboml")
+        return getattr(module, "MetaBackend").load(backend, **kwargs)
+
     else:
         raise_error(
             ValueError,