Migrate python/ directory.

Orbax Authors · Orbax Authors · commit 355e8b5b70dc · 2026-02-11T11:45:30.000-08:00
PiperOrigin-RevId: 868773893
diff --git a/docs/guides/model/orbax_model_mira_runtime_mnist_demo.ipynb b/docs/guides/model/orbax_model_mira_runtime_mnist_demo.ipynb
@@ -601,7 +601,7 @@
       "outputs": [],
       "source": [
         "# Load the model using Orbax Model Runner (new!)\n",
-        "from .learning.infra.mira.experimental.orbax_model.python import orbax_model_runner"
+        "from .neptune.python import orbax_model_runner"
       ]
     },
     {

Original file line number	Diff line number	Diff line change
`@@ -601,7 +601,7 @@`
`601`	`601`	`"outputs": [],`
`602`	`602`	`"source": [`
`603`	`603`	`"# Load the model using Orbax Model Runner (new!)\n",`
`604`		`- "from .learning.infra.mira.experimental.orbax_model.python import orbax_model_runner"`
	`604`	`+ "from .neptune.python import orbax_model_runner"`
`605`	`605`	`]`
`606`	`606`	`},`
`607`	`607`	`{`