Add matplotlib as requirement

pyproject.toml

@@ -13,6 +13,7 @@ dependencies = [
   "numpy>=2",
   "scipy>=1.7",
   "torch>=2",
+  "matplotlib>=3.4",
 ]
 requires-python = ">=3.6"
 authors = [