add MPS warning to two moons notebook

examples/2024-07-19 Two Moons.ipynb

@@ -20,7 +20,9 @@
    "id": "998b6a54389e9e31",
    "metadata": {},
    "source": [
-    "Define which deep learning backend to use. Generally, PyTorch is best for debugging, while JAX is the fastest. See also the installation instructions for BayesFlow."
+    "Define which deep learning backend to use. Generally, PyTorch is best for debugging, while JAX is the fastest. See also the installation instructions for BayesFlow.\n",
+    "\n",
+    "Note that MPS (Apple's acceleration framework) is technically supported, but its operations coverage is still low, so use at your own risk! "
    ]
   },
   {
@@ -454,7 +456,7 @@
    },
    "source": [
     "data = approximator.sample(\n",
-    "    num_samples=1024,\n",
+    "    batch_shape=(1024,),\n",
     "    conditions={\"x\": np.zeros((1024, 2))},\n",
     ")\n",
     "\n",
@@ -512,19 +514,6 @@
     }
    ],
    "execution_count": 14
-  },
-  {
-   "cell_type": "code",
-   "id": "2739498de4152c53",
-   "metadata": {
-    "ExecuteTime": {
-     "end_time": "2024-08-14T15:36:22.772693Z",
-     "start_time": "2024-08-14T15:36:22.771494Z"
-    }
-   },
-   "source": [],
-   "outputs": [],
-   "execution_count": null
   }
  ],
  "metadata": {