diff --git a/docs/pipeline.md b/docs/pipeline.md
new file mode 100644
index 00000000..f69c21c6
--- /dev/null
+++ b/docs/pipeline.md
@@ -0,0 +1,3 @@
+## Polars Native Machine Learning Pipeline
+
+::: polars_ds.pipeline
\ No newline at end of file
diff --git a/docs/sample.md b/docs/sample.md
new file mode 100644
index 00000000..a6a2b493
--- /dev/null
+++ b/docs/sample.md
@@ -0,0 +1,3 @@
+## Polars Native Machine Learning Pipeline
+
+::: polars_ds.sample
\ No newline at end of file
diff --git a/examples/sample_and_split.ipynb b/examples/sample_and_split.ipynb
index a59f765f..3906b586 100644
--- a/examples/sample_and_split.ipynb
+++ b/examples/sample_and_split.ipynb
@@ -42,7 +42,7 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (5, 8)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>row_num</th><th>uniform_1</th><th>uniform_2</th><th>exp</th><th>normal</th><th>fat_normal</th><th>flags</th><th>category</th></tr><tr><td>i64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>i32</td><td>str</td></tr></thead><tbody><tr><td>0</td><td>1.864966</td><td>0.983112</td><td>2.134098</td><td>0.201208</td><td>-510.167817</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>1</td><td>3.681365</td><td>0.072616</td><td>0.24552</td><td>-2.180395</td><td>-1230.797029</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>2</td><td>7.138163</td><td>0.068923</td><td>0.498513</td><td>-1.737763</td><td>-914.757436</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>3</td><td>9.2411</td><td>0.006129</td><td>0.670527</td><td>-1.500905</td><td>-1025.144372</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>4</td><td>7.874972</td><td>0.638764</td><td>2.159589</td><td>0.490217</td><td>1329.546535</td><td>0</td><td>&quot;A&quot;</td></tr></tbody></table></div>"
+       "<small>shape: (5, 8)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>row_num</th><th>uniform_1</th><th>uniform_2</th><th>exp</th><th>normal</th><th>fat_normal</th><th>flags</th><th>category</th></tr><tr><td>i64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>i32</td><td>str</td></tr></thead><tbody><tr><td>0</td><td>0.101301</td><td>0.241236</td><td>0.068629</td><td>-1.546608</td><td>-1820.064986</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>1</td><td>7.763778</td><td>0.688527</td><td>1.564067</td><td>-0.219875</td><td>2842.128922</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>2</td><td>6.692104</td><td>0.302039</td><td>2.184995</td><td>-2.038565</td><td>-107.693276</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>3</td><td>8.069798</td><td>0.589782</td><td>0.047991</td><td>-1.029594</td><td>52.045207</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>4</td><td>4.709925</td><td>0.289922</td><td>0.538721</td><td>1.08094</td><td>711.807154</td><td>1</td><td>&quot;A&quot;</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (5, 8)\n",
@@ -51,11 +51,11 @@
        "│ ---     ┆ ---       ┆ ---       ┆ ---      ┆ ---       ┆ ---          ┆ ---   ┆ ---      │\n",
        "│ i64     ┆ f64       ┆ f64       ┆ f64      ┆ f64       ┆ f64          ┆ i32   ┆ str      │\n",
        "╞═════════╪═══════════╪═══════════╪══════════╪═══════════╪══════════════╪═══════╪══════════╡\n",
-       "│ 0       ┆ 1.864966  ┆ 0.983112  ┆ 2.134098 ┆ 0.201208  ┆ -510.167817  ┆ 1     ┆ A        │\n",
-       "│ 1       ┆ 3.681365  ┆ 0.072616  ┆ 0.24552  ┆ -2.180395 ┆ -1230.797029 ┆ 2     ┆ A        │\n",
-       "│ 2       ┆ 7.138163  ┆ 0.068923  ┆ 0.498513 ┆ -1.737763 ┆ -914.757436  ┆ 2     ┆ A        │\n",
-       "│ 3       ┆ 9.2411    ┆ 0.006129  ┆ 0.670527 ┆ -1.500905 ┆ -1025.144372 ┆ 1     ┆ A        │\n",
-       "│ 4       ┆ 7.874972  ┆ 0.638764  ┆ 2.159589 ┆ 0.490217  ┆ 1329.546535  ┆ 0     ┆ A        │\n",
+       "│ 0       ┆ 0.101301  ┆ 0.241236  ┆ 0.068629 ┆ -1.546608 ┆ -1820.064986 ┆ 0     ┆ A        │\n",
+       "│ 1       ┆ 7.763778  ┆ 0.688527  ┆ 1.564067 ┆ -0.219875 ┆ 2842.128922  ┆ 0     ┆ A        │\n",
+       "│ 2       ┆ 6.692104  ┆ 0.302039  ┆ 2.184995 ┆ -2.038565 ┆ -107.693276  ┆ 0     ┆ A        │\n",
+       "│ 3       ┆ 8.069798  ┆ 0.589782  ┆ 0.047991 ┆ -1.029594 ┆ 52.045207    ┆ 1     ┆ A        │\n",
+       "│ 4       ┆ 4.709925  ┆ 0.289922  ┆ 0.538721 ┆ 1.08094   ┆ 711.807154   ┆ 1     ┆ A        │\n",
        "└─────────┴───────────┴───────────┴──────────┴───────────┴──────────────┴───────┴──────────┘"
       ]
      },
@@ -85,7 +85,7 @@
     {
      "data": {
       "text/plain": [
-       "['row_num', 'fat_normal', 'flags']"
+       "['row_num', 'uniform_1', 'fat_normal']"
       ]
      },
      "execution_count": 3,
@@ -112,27 +112,27 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (60_000, 8)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>row_num</th><th>uniform_1</th><th>uniform_2</th><th>exp</th><th>normal</th><th>fat_normal</th><th>flags</th><th>category</th></tr><tr><td>i64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>i32</td><td>str</td></tr></thead><tbody><tr><td>0</td><td>1.864966</td><td>0.983112</td><td>2.134098</td><td>0.201208</td><td>-510.167817</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>1</td><td>3.681365</td><td>0.072616</td><td>0.24552</td><td>-2.180395</td><td>-1230.797029</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>2</td><td>7.138163</td><td>0.068923</td><td>0.498513</td><td>-1.737763</td><td>-914.757436</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>4</td><td>7.874972</td><td>0.638764</td><td>2.159589</td><td>0.490217</td><td>1329.546535</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>5</td><td>1.479464</td><td>0.703649</td><td>0.48033</td><td>0.550675</td><td>186.879146</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>6</td><td>11.354059</td><td>0.881735</td><td>2.399495</td><td>-0.720839</td><td>-376.389466</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>7</td><td>7.981127</td><td>0.199884</td><td>0.335183</td><td>-0.223638</td><td>-168.936382</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>8</td><td>3.539316</td><td>0.618786</td><td>2.968125</td><td>-0.066433</td><td>3852.290115</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>12</td><td>2.881549</td><td>0.306653</td><td>1.828217</td><td>0.594723</td><td>1189.461612</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>13</td><td>11.286761</td><td>0.200552</td><td>4.836837</td><td>0.725427</td><td>-18.039201</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>16</td><td>7.879337</td><td>0.194351</td><td>1.395224</td><td>-1.621337</td><td>253.488052</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>17</td><td>6.88781</td><td>0.436873</td><td>2.56461</td><td>0.799571</td><td>-573.229085</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td></tr><tr><td>99984</td><td>3.175364</td><td>0.521214</td><td>2.621609</td><td>-0.259841</td><td>-207.322052</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99985</td><td>2.982497</td><td>0.046959</td><td>2.013501</td><td>-1.38318</td><td>-449.496094</td><td>0</td><td>&quot;C&quot;</td></tr><tr><td>99986</td><td>9.715497</td><td>0.173048</td><td>7.787587</td><td>0.031314</td><td>-745.802171</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99987</td><td>0.464149</td><td>0.052577</td><td>0.251207</td><td>-0.550988</td><td>211.994715</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99988</td><td>2.90002</td><td>0.140729</td><td>0.39061</td><td>0.420128</td><td>543.348445</td><td>0</td><td>&quot;C&quot;</td></tr><tr><td>99989</td><td>7.725916</td><td>0.608843</td><td>1.518604</td><td>0.872585</td><td>-1046.055152</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99990</td><td>2.550447</td><td>0.136828</td><td>1.878146</td><td>-0.443894</td><td>1299.166574</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99993</td><td>6.344932</td><td>0.91468</td><td>0.077694</td><td>0.667014</td><td>274.197044</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99995</td><td>7.751136</td><td>0.991215</td><td>0.341938</td><td>0.74724</td><td>468.091164</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99996</td><td>8.932153</td><td>0.149342</td><td>2.017135</td><td>0.254913</td><td>-1555.19188</td><td>0</td><td>&quot;C&quot;</td></tr><tr><td>99997</td><td>6.366458</td><td>0.825137</td><td>1.149404</td><td>-0.653029</td><td>211.288954</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99999</td><td>5.021382</td><td>0.405893</td><td>2.745867</td><td>0.800525</td><td>1572.618783</td><td>1</td><td>&quot;C&quot;</td></tr></tbody></table></div>"
+       "<small>shape: (60_000, 8)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>row_num</th><th>uniform_1</th><th>uniform_2</th><th>exp</th><th>normal</th><th>fat_normal</th><th>flags</th><th>category</th></tr><tr><td>i64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>i32</td><td>str</td></tr></thead><tbody><tr><td>2</td><td>6.692104</td><td>0.302039</td><td>2.184995</td><td>-2.038565</td><td>-107.693276</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>3</td><td>8.069798</td><td>0.589782</td><td>0.047991</td><td>-1.029594</td><td>52.045207</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>4</td><td>4.709925</td><td>0.289922</td><td>0.538721</td><td>1.08094</td><td>711.807154</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>7</td><td>9.679737</td><td>0.431594</td><td>1.131895</td><td>-0.739637</td><td>1269.613996</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>8</td><td>10.803224</td><td>0.546317</td><td>3.25459</td><td>-0.904231</td><td>243.572733</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td></tr><tr><td>99991</td><td>10.25385</td><td>0.632615</td><td>0.263188</td><td>-0.388282</td><td>273.417495</td><td>2</td><td>&quot;C&quot;</td></tr><tr><td>99993</td><td>5.213348</td><td>0.199494</td><td>4.928223</td><td>-0.362607</td><td>1729.126716</td><td>0</td><td>&quot;C&quot;</td></tr><tr><td>99994</td><td>9.441603</td><td>0.347907</td><td>0.318096</td><td>-0.112797</td><td>242.457284</td><td>0</td><td>&quot;C&quot;</td></tr><tr><td>99998</td><td>7.374466</td><td>0.030731</td><td>3.606166</td><td>-0.582265</td><td>1290.937356</td><td>0</td><td>&quot;C&quot;</td></tr><tr><td>99999</td><td>7.164853</td><td>0.399791</td><td>0.354686</td><td>-0.999868</td><td>1678.190405</td><td>1</td><td>&quot;C&quot;</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (60_000, 8)\n",
-       "┌─────────┬───────────┬───────────┬──────────┬───────────┬──────────────┬───────┬──────────┐\n",
-       "│ row_num ┆ uniform_1 ┆ uniform_2 ┆ exp      ┆ normal    ┆ fat_normal   ┆ flags ┆ category │\n",
-       "│ ---     ┆ ---       ┆ ---       ┆ ---      ┆ ---       ┆ ---          ┆ ---   ┆ ---      │\n",
-       "│ i64     ┆ f64       ┆ f64       ┆ f64      ┆ f64       ┆ f64          ┆ i32   ┆ str      │\n",
-       "╞═════════╪═══════════╪═══════════╪══════════╪═══════════╪══════════════╪═══════╪══════════╡\n",
-       "│ 0       ┆ 1.864966  ┆ 0.983112  ┆ 2.134098 ┆ 0.201208  ┆ -510.167817  ┆ 1     ┆ A        │\n",
-       "│ 1       ┆ 3.681365  ┆ 0.072616  ┆ 0.24552  ┆ -2.180395 ┆ -1230.797029 ┆ 2     ┆ A        │\n",
-       "│ 2       ┆ 7.138163  ┆ 0.068923  ┆ 0.498513 ┆ -1.737763 ┆ -914.757436  ┆ 2     ┆ A        │\n",
-       "│ 4       ┆ 7.874972  ┆ 0.638764  ┆ 2.159589 ┆ 0.490217  ┆ 1329.546535  ┆ 0     ┆ A        │\n",
-       "│ 5       ┆ 1.479464  ┆ 0.703649  ┆ 0.48033  ┆ 0.550675  ┆ 186.879146   ┆ 1     ┆ A        │\n",
-       "│ …       ┆ …         ┆ …         ┆ …        ┆ …         ┆ …            ┆ …     ┆ …        │\n",
-       "│ 99993   ┆ 6.344932  ┆ 0.91468   ┆ 0.077694 ┆ 0.667014  ┆ 274.197044   ┆ 1     ┆ C        │\n",
-       "│ 99995   ┆ 7.751136  ┆ 0.991215  ┆ 0.341938 ┆ 0.74724   ┆ 468.091164   ┆ 1     ┆ C        │\n",
-       "│ 99996   ┆ 8.932153  ┆ 0.149342  ┆ 2.017135 ┆ 0.254913  ┆ -1555.19188  ┆ 0     ┆ C        │\n",
-       "│ 99997   ┆ 6.366458  ┆ 0.825137  ┆ 1.149404 ┆ -0.653029 ┆ 211.288954   ┆ 1     ┆ C        │\n",
-       "│ 99999   ┆ 5.021382  ┆ 0.405893  ┆ 2.745867 ┆ 0.800525  ┆ 1572.618783  ┆ 1     ┆ C        │\n",
-       "└─────────┴───────────┴───────────┴──────────┴───────────┴──────────────┴───────┴──────────┘"
+       "┌─────────┬───────────┬───────────┬──────────┬───────────┬─────────────┬───────┬──────────┐\n",
+       "│ row_num ┆ uniform_1 ┆ uniform_2 ┆ exp      ┆ normal    ┆ fat_normal  ┆ flags ┆ category │\n",
+       "│ ---     ┆ ---       ┆ ---       ┆ ---      ┆ ---       ┆ ---         ┆ ---   ┆ ---      │\n",
+       "│ i64     ┆ f64       ┆ f64       ┆ f64      ┆ f64       ┆ f64         ┆ i32   ┆ str      │\n",
+       "╞═════════╪═══════════╪═══════════╪══════════╪═══════════╪═════════════╪═══════╪══════════╡\n",
+       "│ 2       ┆ 6.692104  ┆ 0.302039  ┆ 2.184995 ┆ -2.038565 ┆ -107.693276 ┆ 0     ┆ A        │\n",
+       "│ 3       ┆ 8.069798  ┆ 0.589782  ┆ 0.047991 ┆ -1.029594 ┆ 52.045207   ┆ 1     ┆ A        │\n",
+       "│ 4       ┆ 4.709925  ┆ 0.289922  ┆ 0.538721 ┆ 1.08094   ┆ 711.807154  ┆ 1     ┆ A        │\n",
+       "│ 7       ┆ 9.679737  ┆ 0.431594  ┆ 1.131895 ┆ -0.739637 ┆ 1269.613996 ┆ 1     ┆ A        │\n",
+       "│ 8       ┆ 10.803224 ┆ 0.546317  ┆ 3.25459  ┆ -0.904231 ┆ 243.572733  ┆ 2     ┆ A        │\n",
+       "│ …       ┆ …         ┆ …         ┆ …        ┆ …         ┆ …           ┆ …     ┆ …        │\n",
+       "│ 99991   ┆ 10.25385  ┆ 0.632615  ┆ 0.263188 ┆ -0.388282 ┆ 273.417495  ┆ 2     ┆ C        │\n",
+       "│ 99993   ┆ 5.213348  ┆ 0.199494  ┆ 4.928223 ┆ -0.362607 ┆ 1729.126716 ┆ 0     ┆ C        │\n",
+       "│ 99994   ┆ 9.441603  ┆ 0.347907  ┆ 0.318096 ┆ -0.112797 ┆ 242.457284  ┆ 0     ┆ C        │\n",
+       "│ 99998   ┆ 7.374466  ┆ 0.030731  ┆ 3.606166 ┆ -0.582265 ┆ 1290.937356 ┆ 0     ┆ C        │\n",
+       "│ 99999   ┆ 7.164853  ┆ 0.399791  ┆ 0.354686 ┆ -0.999868 ┆ 1678.190405 ┆ 1     ┆ C        │\n",
+       "└─────────┴───────────┴───────────┴──────────┴───────────┴─────────────┴───────┴──────────┘"
       ]
      },
      "execution_count": 4,
@@ -160,27 +160,27 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (30_000, 8)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>row_num</th><th>uniform_1</th><th>uniform_2</th><th>exp</th><th>normal</th><th>fat_normal</th><th>flags</th><th>category</th></tr><tr><td>i64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>i32</td><td>str</td></tr></thead><tbody><tr><td>4</td><td>7.874972</td><td>0.638764</td><td>2.159589</td><td>0.490217</td><td>1329.546535</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>5</td><td>1.479464</td><td>0.703649</td><td>0.48033</td><td>0.550675</td><td>186.879146</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>6</td><td>11.354059</td><td>0.881735</td><td>2.399495</td><td>-0.720839</td><td>-376.389466</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>8</td><td>3.539316</td><td>0.618786</td><td>2.968125</td><td>-0.066433</td><td>3852.290115</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>12</td><td>2.881549</td><td>0.306653</td><td>1.828217</td><td>0.594723</td><td>1189.461612</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>13</td><td>11.286761</td><td>0.200552</td><td>4.836837</td><td>0.725427</td><td>-18.039201</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>22</td><td>2.833185</td><td>0.043957</td><td>3.999065</td><td>-0.735753</td><td>357.525422</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>25</td><td>11.359116</td><td>0.191183</td><td>0.703892</td><td>0.195382</td><td>-1028.061174</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>27</td><td>6.276564</td><td>0.892571</td><td>0.734094</td><td>0.648778</td><td>-1995.465652</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>32</td><td>11.959182</td><td>0.665287</td><td>2.574619</td><td>1.856659</td><td>-982.271182</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>34</td><td>6.493756</td><td>0.947386</td><td>1.745573</td><td>0.261902</td><td>598.127735</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>41</td><td>9.549829</td><td>0.69802</td><td>2.068939</td><td>-1.506053</td><td>57.884184</td><td>2</td><td>&quot;A&quot;</td></tr><tr><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td></tr><tr><td>99965</td><td>8.612262</td><td>0.692854</td><td>3.455756</td><td>0.779895</td><td>433.966147</td><td>2</td><td>&quot;C&quot;</td></tr><tr><td>99966</td><td>3.343289</td><td>0.487011</td><td>3.656845</td><td>0.91173</td><td>870.313838</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99968</td><td>4.45978</td><td>0.079741</td><td>3.194479</td><td>-0.210791</td><td>-843.718407</td><td>2</td><td>&quot;C&quot;</td></tr><tr><td>99971</td><td>9.711985</td><td>0.050049</td><td>0.078004</td><td>1.655786</td><td>-1371.070708</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99972</td><td>1.047484</td><td>0.808619</td><td>0.347894</td><td>0.834854</td><td>-1020.64733</td><td>2</td><td>&quot;C&quot;</td></tr><tr><td>99977</td><td>8.589407</td><td>0.535741</td><td>0.131577</td><td>1.839748</td><td>-104.372155</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99980</td><td>9.513081</td><td>0.958727</td><td>1.197324</td><td>2.042775</td><td>-1971.061272</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99982</td><td>2.048097</td><td>0.19573</td><td>1.607711</td><td>0.141</td><td>-525.757109</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99984</td><td>3.175364</td><td>0.521214</td><td>2.621609</td><td>-0.259841</td><td>-207.322052</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99991</td><td>0.155865</td><td>0.938876</td><td>1.293072</td><td>-3.370048</td><td>677.127453</td><td>2</td><td>&quot;C&quot;</td></tr><tr><td>99995</td><td>7.751136</td><td>0.991215</td><td>0.341938</td><td>0.74724</td><td>468.091164</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99997</td><td>6.366458</td><td>0.825137</td><td>1.149404</td><td>-0.653029</td><td>211.288954</td><td>1</td><td>&quot;C&quot;</td></tr></tbody></table></div>"
+       "<small>shape: (30_000, 8)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>row_num</th><th>uniform_1</th><th>uniform_2</th><th>exp</th><th>normal</th><th>fat_normal</th><th>flags</th><th>category</th></tr><tr><td>i64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>f64</td><td>i32</td><td>str</td></tr></thead><tbody><tr><td>14</td><td>3.29052</td><td>0.62766</td><td>0.684437</td><td>-0.114606</td><td>1440.324971</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>15</td><td>10.033849</td><td>0.555832</td><td>0.265698</td><td>1.894409</td><td>-2266.65647</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>18</td><td>10.209092</td><td>0.411413</td><td>1.781309</td><td>1.502181</td><td>63.521515</td><td>1</td><td>&quot;A&quot;</td></tr><tr><td>23</td><td>5.461989</td><td>0.631279</td><td>2.069071</td><td>-0.735686</td><td>-1463.48885</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>27</td><td>8.902278</td><td>0.732549</td><td>2.877614</td><td>-0.049382</td><td>-1301.745218</td><td>0</td><td>&quot;A&quot;</td></tr><tr><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td><td>&hellip;</td></tr><tr><td>99974</td><td>6.541671</td><td>0.052039</td><td>2.622168</td><td>-0.451139</td><td>-1270.959364</td><td>2</td><td>&quot;C&quot;</td></tr><tr><td>99976</td><td>1.147642</td><td>0.885223</td><td>0.704189</td><td>0.61381</td><td>344.861659</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99985</td><td>7.321052</td><td>0.76491</td><td>6.039978</td><td>-0.407754</td><td>304.340042</td><td>1</td><td>&quot;C&quot;</td></tr><tr><td>99991</td><td>10.25385</td><td>0.632615</td><td>0.263188</td><td>-0.388282</td><td>273.417495</td><td>2</td><td>&quot;C&quot;</td></tr><tr><td>99995</td><td>1.948428</td><td>0.923293</td><td>6.168104</td><td>-0.151161</td><td>997.159514</td><td>0</td><td>&quot;C&quot;</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (30_000, 8)\n",
-       "┌─────────┬───────────┬───────────┬──────────┬───────────┬─────────────┬───────┬──────────┐\n",
-       "│ row_num ┆ uniform_1 ┆ uniform_2 ┆ exp      ┆ normal    ┆ fat_normal  ┆ flags ┆ category │\n",
-       "│ ---     ┆ ---       ┆ ---       ┆ ---      ┆ ---       ┆ ---         ┆ ---   ┆ ---      │\n",
-       "│ i64     ┆ f64       ┆ f64       ┆ f64      ┆ f64       ┆ f64         ┆ i32   ┆ str      │\n",
-       "╞═════════╪═══════════╪═══════════╪══════════╪═══════════╪═════════════╪═══════╪══════════╡\n",
-       "│ 4       ┆ 7.874972  ┆ 0.638764  ┆ 2.159589 ┆ 0.490217  ┆ 1329.546535 ┆ 0     ┆ A        │\n",
-       "│ 5       ┆ 1.479464  ┆ 0.703649  ┆ 0.48033  ┆ 0.550675  ┆ 186.879146  ┆ 1     ┆ A        │\n",
-       "│ 6       ┆ 11.354059 ┆ 0.881735  ┆ 2.399495 ┆ -0.720839 ┆ -376.389466 ┆ 2     ┆ A        │\n",
-       "│ 8       ┆ 3.539316  ┆ 0.618786  ┆ 2.968125 ┆ -0.066433 ┆ 3852.290115 ┆ 0     ┆ A        │\n",
-       "│ 12      ┆ 2.881549  ┆ 0.306653  ┆ 1.828217 ┆ 0.594723  ┆ 1189.461612 ┆ 0     ┆ A        │\n",
-       "│ …       ┆ …         ┆ …         ┆ …        ┆ …         ┆ …           ┆ …     ┆ …        │\n",
-       "│ 99982   ┆ 2.048097  ┆ 0.19573   ┆ 1.607711 ┆ 0.141     ┆ -525.757109 ┆ 1     ┆ C        │\n",
-       "│ 99984   ┆ 3.175364  ┆ 0.521214  ┆ 2.621609 ┆ -0.259841 ┆ -207.322052 ┆ 1     ┆ C        │\n",
-       "│ 99991   ┆ 0.155865  ┆ 0.938876  ┆ 1.293072 ┆ -3.370048 ┆ 677.127453  ┆ 2     ┆ C        │\n",
-       "│ 99995   ┆ 7.751136  ┆ 0.991215  ┆ 0.341938 ┆ 0.74724   ┆ 468.091164  ┆ 1     ┆ C        │\n",
-       "│ 99997   ┆ 6.366458  ┆ 0.825137  ┆ 1.149404 ┆ -0.653029 ┆ 211.288954  ┆ 1     ┆ C        │\n",
-       "└─────────┴───────────┴───────────┴──────────┴───────────┴─────────────┴───────┴──────────┘"
+       "┌─────────┬───────────┬───────────┬──────────┬───────────┬──────────────┬───────┬──────────┐\n",
+       "│ row_num ┆ uniform_1 ┆ uniform_2 ┆ exp      ┆ normal    ┆ fat_normal   ┆ flags ┆ category │\n",
+       "│ ---     ┆ ---       ┆ ---       ┆ ---      ┆ ---       ┆ ---          ┆ ---   ┆ ---      │\n",
+       "│ i64     ┆ f64       ┆ f64       ┆ f64      ┆ f64       ┆ f64          ┆ i32   ┆ str      │\n",
+       "╞═════════╪═══════════╪═══════════╪══════════╪═══════════╪══════════════╪═══════╪══════════╡\n",
+       "│ 14      ┆ 3.29052   ┆ 0.62766   ┆ 0.684437 ┆ -0.114606 ┆ 1440.324971  ┆ 0     ┆ A        │\n",
+       "│ 15      ┆ 10.033849 ┆ 0.555832  ┆ 0.265698 ┆ 1.894409  ┆ -2266.65647  ┆ 1     ┆ A        │\n",
+       "│ 18      ┆ 10.209092 ┆ 0.411413  ┆ 1.781309 ┆ 1.502181  ┆ 63.521515    ┆ 1     ┆ A        │\n",
+       "│ 23      ┆ 5.461989  ┆ 0.631279  ┆ 2.069071 ┆ -0.735686 ┆ -1463.48885  ┆ 0     ┆ A        │\n",
+       "│ 27      ┆ 8.902278  ┆ 0.732549  ┆ 2.877614 ┆ -0.049382 ┆ -1301.745218 ┆ 0     ┆ A        │\n",
+       "│ …       ┆ …         ┆ …         ┆ …        ┆ …         ┆ …            ┆ …     ┆ …        │\n",
+       "│ 99974   ┆ 6.541671  ┆ 0.052039  ┆ 2.622168 ┆ -0.451139 ┆ -1270.959364 ┆ 2     ┆ C        │\n",
+       "│ 99976   ┆ 1.147642  ┆ 0.885223  ┆ 0.704189 ┆ 0.61381   ┆ 344.861659   ┆ 1     ┆ C        │\n",
+       "│ 99985   ┆ 7.321052  ┆ 0.76491   ┆ 6.039978 ┆ -0.407754 ┆ 304.340042   ┆ 1     ┆ C        │\n",
+       "│ 99991   ┆ 10.25385  ┆ 0.632615  ┆ 0.263188 ┆ -0.388282 ┆ 273.417495   ┆ 2     ┆ C        │\n",
+       "│ 99995   ┆ 1.948428  ┆ 0.923293  ┆ 6.168104 ┆ -0.151161 ┆ 997.159514   ┆ 0     ┆ C        │\n",
+       "└─────────┴───────────┴───────────┴──────────┴───────────┴──────────────┴───────┴──────────┘"
       ]
      },
      "execution_count": 5,
@@ -207,7 +207,7 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>flags</th><th>len</th></tr><tr><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>0</td><td>33281</td></tr><tr><td>1</td><td>33469</td></tr><tr><td>2</td><td>33250</td></tr></tbody></table></div>"
+       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>flags</th><th>len</th></tr><tr><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>0</td><td>33282</td></tr><tr><td>1</td><td>33654</td></tr><tr><td>2</td><td>33064</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (3, 2)\n",
@@ -216,9 +216,9 @@
        "│ ---   ┆ ---   │\n",
        "│ i32   ┆ u32   │\n",
        "╞═══════╪═══════╡\n",
-       "│ 0     ┆ 33281 │\n",
-       "│ 1     ┆ 33469 │\n",
-       "│ 2     ┆ 33250 │\n",
+       "│ 0     ┆ 33282 │\n",
+       "│ 1     ┆ 33654 │\n",
+       "│ 2     ┆ 33064 │\n",
        "└───────┴───────┘"
       ]
      },
@@ -246,7 +246,7 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>flags</th><th>len</th></tr><tr><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>0</td><td>16641</td></tr><tr><td>1</td><td>33469</td></tr><tr><td>2</td><td>33250</td></tr></tbody></table></div>"
+       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>flags</th><th>len</th></tr><tr><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>0</td><td>16641</td></tr><tr><td>1</td><td>33654</td></tr><tr><td>2</td><td>33064</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (3, 2)\n",
@@ -256,8 +256,8 @@
        "│ i32   ┆ u32   │\n",
        "╞═══════╪═══════╡\n",
        "│ 0     ┆ 16641 │\n",
-       "│ 1     ┆ 33469 │\n",
-       "│ 2     ┆ 33250 │\n",
+       "│ 1     ┆ 33654 │\n",
+       "│ 2     ┆ 33064 │\n",
        "└───────┴───────┘"
       ]
      },
@@ -290,7 +290,7 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>flags</th><th>len</th></tr><tr><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>0</td><td>16641</td></tr><tr><td>1</td><td>10041</td></tr><tr><td>2</td><td>13300</td></tr></tbody></table></div>"
+       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>flags</th><th>len</th></tr><tr><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>0</td><td>16641</td></tr><tr><td>1</td><td>10097</td></tr><tr><td>2</td><td>13226</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (3, 2)\n",
@@ -300,8 +300,8 @@
        "│ i32   ┆ u32   │\n",
        "╞═══════╪═══════╡\n",
        "│ 0     ┆ 16641 │\n",
-       "│ 1     ┆ 10041 │\n",
-       "│ 2     ┆ 13300 │\n",
+       "│ 1     ┆ 10097 │\n",
+       "│ 2     ┆ 13226 │\n",
        "└───────┴───────┘"
       ]
      },
@@ -466,7 +466,7 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>category</th><th>len</th></tr><tr><td>str</td><td>u32</td></tr></thead><tbody><tr><td>&quot;A&quot;</td><td>10000</td></tr><tr><td>&quot;B&quot;</td><td>4249</td></tr><tr><td>&quot;C&quot;</td><td>5751</td></tr></tbody></table></div>"
+       "<small>shape: (3, 2)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>category</th><th>len</th></tr><tr><td>str</td><td>u32</td></tr></thead><tbody><tr><td>&quot;A&quot;</td><td>10000</td></tr><tr><td>&quot;B&quot;</td><td>4302</td></tr><tr><td>&quot;C&quot;</td><td>5698</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (3, 2)\n",
@@ -476,8 +476,8 @@
        "│ str      ┆ u32   │\n",
        "╞══════════╪═══════╡\n",
        "│ A        ┆ 10000 │\n",
-       "│ B        ┆ 4249  │\n",
-       "│ C        ┆ 5751  │\n",
+       "│ B        ┆ 4302  │\n",
+       "│ C        ┆ 5698  │\n",
        "└──────────┴───────┘"
       ]
      },
@@ -511,7 +511,7 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (9, 3)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>category</th><th>flags</th><th>len</th></tr><tr><td>str</td><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>&quot;A&quot;</td><td>0</td><td>9996</td></tr><tr><td>&quot;A&quot;</td><td>1</td><td>9996</td></tr><tr><td>&quot;A&quot;</td><td>2</td><td>9996</td></tr><tr><td>&quot;B&quot;</td><td>0</td><td>9970</td></tr><tr><td>&quot;B&quot;</td><td>1</td><td>9970</td></tr><tr><td>&quot;B&quot;</td><td>2</td><td>9970</td></tr><tr><td>&quot;C&quot;</td><td>0</td><td>13251</td></tr><tr><td>&quot;C&quot;</td><td>1</td><td>13251</td></tr><tr><td>&quot;C&quot;</td><td>2</td><td>13251</td></tr></tbody></table></div>"
+       "<small>shape: (9, 3)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>category</th><th>flags</th><th>len</th></tr><tr><td>str</td><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>&quot;A&quot;</td><td>0</td><td>9865</td></tr><tr><td>&quot;A&quot;</td><td>1</td><td>9865</td></tr><tr><td>&quot;A&quot;</td><td>2</td><td>9865</td></tr><tr><td>&quot;B&quot;</td><td>0</td><td>9909</td></tr><tr><td>&quot;B&quot;</td><td>1</td><td>9909</td></tr><tr><td>&quot;B&quot;</td><td>2</td><td>9909</td></tr><tr><td>&quot;C&quot;</td><td>0</td><td>13224</td></tr><tr><td>&quot;C&quot;</td><td>1</td><td>13224</td></tr><tr><td>&quot;C&quot;</td><td>2</td><td>13224</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (9, 3)\n",
@@ -520,15 +520,15 @@
        "│ ---      ┆ ---   ┆ ---   │\n",
        "│ str      ┆ i32   ┆ u32   │\n",
        "╞══════════╪═══════╪═══════╡\n",
-       "│ A        ┆ 0     ┆ 9996  │\n",
-       "│ A        ┆ 1     ┆ 9996  │\n",
-       "│ A        ┆ 2     ┆ 9996  │\n",
-       "│ B        ┆ 0     ┆ 9970  │\n",
-       "│ B        ┆ 1     ┆ 9970  │\n",
-       "│ B        ┆ 2     ┆ 9970  │\n",
-       "│ C        ┆ 0     ┆ 13251 │\n",
-       "│ C        ┆ 1     ┆ 13251 │\n",
-       "│ C        ┆ 2     ┆ 13251 │\n",
+       "│ A        ┆ 0     ┆ 9865  │\n",
+       "│ A        ┆ 1     ┆ 9865  │\n",
+       "│ A        ┆ 2     ┆ 9865  │\n",
+       "│ B        ┆ 0     ┆ 9909  │\n",
+       "│ B        ┆ 1     ┆ 9909  │\n",
+       "│ B        ┆ 2     ┆ 9909  │\n",
+       "│ C        ┆ 0     ┆ 13224 │\n",
+       "│ C        ┆ 1     ┆ 13224 │\n",
+       "│ C        ┆ 2     ┆ 13224 │\n",
        "└──────────┴───────┴───────┘"
       ]
      },
@@ -563,7 +563,7 @@
        "  white-space: pre-wrap;\n",
        "}\n",
        "</style>\n",
-       "<small>shape: (9, 3)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>category</th><th>flags</th><th>len</th></tr><tr><td>str</td><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>&quot;A&quot;</td><td>0</td><td>9996</td></tr><tr><td>&quot;A&quot;</td><td>1</td><td>9996</td></tr><tr><td>&quot;A&quot;</td><td>2</td><td>9996</td></tr><tr><td>&quot;B&quot;</td><td>0</td><td>9970</td></tr><tr><td>&quot;B&quot;</td><td>1</td><td>9970</td></tr><tr><td>&quot;B&quot;</td><td>2</td><td>9970</td></tr><tr><td>&quot;C&quot;</td><td>0</td><td>10000</td></tr><tr><td>&quot;C&quot;</td><td>1</td><td>10000</td></tr><tr><td>&quot;C&quot;</td><td>2</td><td>10000</td></tr></tbody></table></div>"
+       "<small>shape: (9, 3)</small><table border=\"1\" class=\"dataframe\"><thead><tr><th>category</th><th>flags</th><th>len</th></tr><tr><td>str</td><td>i32</td><td>u32</td></tr></thead><tbody><tr><td>&quot;A&quot;</td><td>0</td><td>9865</td></tr><tr><td>&quot;A&quot;</td><td>1</td><td>9865</td></tr><tr><td>&quot;A&quot;</td><td>2</td><td>9865</td></tr><tr><td>&quot;B&quot;</td><td>0</td><td>9909</td></tr><tr><td>&quot;B&quot;</td><td>1</td><td>9909</td></tr><tr><td>&quot;B&quot;</td><td>2</td><td>9909</td></tr><tr><td>&quot;C&quot;</td><td>0</td><td>10000</td></tr><tr><td>&quot;C&quot;</td><td>1</td><td>10000</td></tr><tr><td>&quot;C&quot;</td><td>2</td><td>10000</td></tr></tbody></table></div>"
       ],
       "text/plain": [
        "shape: (9, 3)\n",
@@ -572,12 +572,12 @@
        "│ ---      ┆ ---   ┆ ---   │\n",
        "│ str      ┆ i32   ┆ u32   │\n",
        "╞══════════╪═══════╪═══════╡\n",
-       "│ A        ┆ 0     ┆ 9996  │\n",
-       "│ A        ┆ 1     ┆ 9996  │\n",
-       "│ A        ┆ 2     ┆ 9996  │\n",
-       "│ B        ┆ 0     ┆ 9970  │\n",
-       "│ B        ┆ 1     ┆ 9970  │\n",
-       "│ B        ┆ 2     ┆ 9970  │\n",
+       "│ A        ┆ 0     ┆ 9865  │\n",
+       "│ A        ┆ 1     ┆ 9865  │\n",
+       "│ A        ┆ 2     ┆ 9865  │\n",
+       "│ B        ┆ 0     ┆ 9909  │\n",
+       "│ B        ┆ 1     ┆ 9909  │\n",
+       "│ B        ┆ 2     ┆ 9909  │\n",
        "│ C        ┆ 0     ┆ 10000 │\n",
        "│ C        ┆ 1     ┆ 10000 │\n",
        "│ C        ┆ 2     ┆ 10000 │\n",
diff --git a/mkdocs.yml b/mkdocs.yml
index e751aa33..190fd62c 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -6,6 +6,8 @@ use_directory_urls: false
 nav:
 - Home: index.md
 - Diagnosis: dia.md
+- Pipeline: pipeline.md
+- Sample: sample.md
 - Numerical Extension: num.md
 - Stats Extension: stats.md
 - String Extension: str2.md
diff --git a/python/polars_ds/sample.py b/python/polars_ds/sample.py
index ca44c815..469ee9f5 100644
--- a/python/polars_ds/sample.py
+++ b/python/polars_ds/sample.py
@@ -47,9 +47,7 @@ def volume_neutral(
     seed: Optional[int] = None,
 ) -> pl.DataFrame:
     """
-    Say we have a reference column, which is discrete. Let's say it has three distinct values, A,
-    B, and C, with a, b, c being the value counts. It will randomly select min(a, b, c, target_volume)
-    rows from each category, thus the name volume neutral.
+    Select volume neutral many population from each segment in `by`, with optional control categories.
 
     Parameters
     ----------
@@ -62,7 +60,7 @@ def volume_neutral(
         Additional level(s). If not none, the volume neutral selection will happen at the
         sublevel of the control column(s). See example.
     target_volume
-        If none, it will select min(a, b, c) rows, this means that one group is always fully selected.
+        If none, it will select min(a, b, c) rows.
     seed
         A random seed
     """