add end-to-end pi0 reproduction notebook

2026-07-12 21:30:10 +08:00
parent 4fe0099f3f
commit 462c070dd9
4 changed files with 639 additions and 0 deletions
@@ -19,6 +19,11 @@ dependencies = [

 [project.optional-dependencies]
 dev = ["pytest>=8.0"]
+notebook = [
+  "ipykernel>=6.29",
+  "jupyterlab>=4.2",
+  "nbconvert>=7.16",
+]

 [project.scripts]
 pi0mini-train = "pi0mini.train:main"