diff --git a/dev/.buildinfo b/dev/.buildinfo
index 7ea3cda..65190af 100644
--- a/dev/.buildinfo
+++ b/dev/.buildinfo
@@ -1,4 +1,4 @@
# Sphinx build info version 1
# This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: 44da94e5d404514dbcaa7ecec9095697
+config: ac510b618caa4214583a0f91a126490d
tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/dev/_autosummary/jaxopt.AndersonAcceleration.html b/dev/_autosummary/jaxopt.AndersonAcceleration.html
index 3a2841a..9830cdd 100644
--- a/dev/_autosummary/jaxopt.AndersonAcceleration.html
+++ b/dev/_autosummary/jaxopt.AndersonAcceleration.html
@@ -4,7 +4,7 @@
-
jaxopt.AndersonAcceleration — JAXopt 0.8 documentation
+ jaxopt.AndersonAcceleration — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.AndersonWrapper.html b/dev/_autosummary/jaxopt.AndersonWrapper.html
index 430548c..4a7d812 100644
--- a/dev/_autosummary/jaxopt.AndersonWrapper.html
+++ b/dev/_autosummary/jaxopt.AndersonWrapper.html
@@ -4,7 +4,7 @@
- jaxopt.AndersonWrapper — JAXopt 0.8 documentation
+ jaxopt.AndersonWrapper — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.ArmijoSGD.html b/dev/_autosummary/jaxopt.ArmijoSGD.html
index 594458a..62d2762 100644
--- a/dev/_autosummary/jaxopt.ArmijoSGD.html
+++ b/dev/_autosummary/jaxopt.ArmijoSGD.html
@@ -4,7 +4,7 @@
- jaxopt.ArmijoSGD — JAXopt 0.8 documentation
+ jaxopt.ArmijoSGD — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.BFGS.html b/dev/_autosummary/jaxopt.BFGS.html
index 7f5386d..7a9a0df 100644
--- a/dev/_autosummary/jaxopt.BFGS.html
+++ b/dev/_autosummary/jaxopt.BFGS.html
@@ -4,7 +4,7 @@
- jaxopt.BFGS — JAXopt 0.8 documentation
+ jaxopt.BFGS — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.BacktrackingLineSearch.html b/dev/_autosummary/jaxopt.BacktrackingLineSearch.html
index 76f17fa..da4139c 100644
--- a/dev/_autosummary/jaxopt.BacktrackingLineSearch.html
+++ b/dev/_autosummary/jaxopt.BacktrackingLineSearch.html
@@ -4,7 +4,7 @@
- jaxopt.BacktrackingLineSearch — JAXopt 0.8 documentation
+ jaxopt.BacktrackingLineSearch — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.Bisection.html b/dev/_autosummary/jaxopt.Bisection.html
index 487bd0b..d6fd040 100644
--- a/dev/_autosummary/jaxopt.Bisection.html
+++ b/dev/_autosummary/jaxopt.Bisection.html
@@ -4,7 +4,7 @@
- jaxopt.Bisection — JAXopt 0.8 documentation
+ jaxopt.Bisection — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.BlockCoordinateDescent.html b/dev/_autosummary/jaxopt.BlockCoordinateDescent.html
index 425dbc2..0259621 100644
--- a/dev/_autosummary/jaxopt.BlockCoordinateDescent.html
+++ b/dev/_autosummary/jaxopt.BlockCoordinateDescent.html
@@ -4,7 +4,7 @@
- jaxopt.BlockCoordinateDescent — JAXopt 0.8 documentation
+ jaxopt.BlockCoordinateDescent — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.BoxCDQP.html b/dev/_autosummary/jaxopt.BoxCDQP.html
index be33575..1b90d26 100644
--- a/dev/_autosummary/jaxopt.BoxCDQP.html
+++ b/dev/_autosummary/jaxopt.BoxCDQP.html
@@ -4,7 +4,7 @@
- jaxopt.BoxCDQP — JAXopt 0.8 documentation
+ jaxopt.BoxCDQP — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.BoxOSQP.html b/dev/_autosummary/jaxopt.BoxOSQP.html
index c94bf50..5e95a0c 100644
--- a/dev/_autosummary/jaxopt.BoxOSQP.html
+++ b/dev/_autosummary/jaxopt.BoxOSQP.html
@@ -4,7 +4,7 @@
- jaxopt.BoxOSQP — JAXopt 0.8 documentation
+ jaxopt.BoxOSQP — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.Broyden.html b/dev/_autosummary/jaxopt.Broyden.html
index bacc983..6338c02 100644
--- a/dev/_autosummary/jaxopt.Broyden.html
+++ b/dev/_autosummary/jaxopt.Broyden.html
@@ -4,7 +4,7 @@
- jaxopt.Broyden — JAXopt 0.8 documentation
+ jaxopt.Broyden — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.CvxpyQP.html b/dev/_autosummary/jaxopt.CvxpyQP.html
index b1ff951..db6b1c1 100644
--- a/dev/_autosummary/jaxopt.CvxpyQP.html
+++ b/dev/_autosummary/jaxopt.CvxpyQP.html
@@ -4,7 +4,7 @@
- jaxopt.CvxpyQP — JAXopt 0.8 documentation
+ jaxopt.CvxpyQP — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.EqualityConstrainedQP.html b/dev/_autosummary/jaxopt.EqualityConstrainedQP.html
index 12a1035..55ab0d7 100644
--- a/dev/_autosummary/jaxopt.EqualityConstrainedQP.html
+++ b/dev/_autosummary/jaxopt.EqualityConstrainedQP.html
@@ -4,7 +4,7 @@
- jaxopt.EqualityConstrainedQP — JAXopt 0.8 documentation
+ jaxopt.EqualityConstrainedQP — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.FixedPointIteration.html b/dev/_autosummary/jaxopt.FixedPointIteration.html
index dec61d8..76b204f 100644
--- a/dev/_autosummary/jaxopt.FixedPointIteration.html
+++ b/dev/_autosummary/jaxopt.FixedPointIteration.html
@@ -4,7 +4,7 @@
- jaxopt.FixedPointIteration — JAXopt 0.8 documentation
+ jaxopt.FixedPointIteration — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.GaussNewton.html b/dev/_autosummary/jaxopt.GaussNewton.html
index a16c491..4434c73 100644
--- a/dev/_autosummary/jaxopt.GaussNewton.html
+++ b/dev/_autosummary/jaxopt.GaussNewton.html
@@ -4,7 +4,7 @@
- jaxopt.GaussNewton — JAXopt 0.8 documentation
+ jaxopt.GaussNewton — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.GradientDescent.html b/dev/_autosummary/jaxopt.GradientDescent.html
index f4dfdca..c9ceece 100644
--- a/dev/_autosummary/jaxopt.GradientDescent.html
+++ b/dev/_autosummary/jaxopt.GradientDescent.html
@@ -4,7 +4,7 @@
- jaxopt.GradientDescent — JAXopt 0.8 documentation
+ jaxopt.GradientDescent — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.HagerZhangLineSearch.html b/dev/_autosummary/jaxopt.HagerZhangLineSearch.html
index 12bd56f..656c011 100644
--- a/dev/_autosummary/jaxopt.HagerZhangLineSearch.html
+++ b/dev/_autosummary/jaxopt.HagerZhangLineSearch.html
@@ -4,7 +4,7 @@
- jaxopt.HagerZhangLineSearch — JAXopt 0.8 documentation
+ jaxopt.HagerZhangLineSearch — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.IterativeRefinement.html b/dev/_autosummary/jaxopt.IterativeRefinement.html
index 5b5ea90..b7decf9 100644
--- a/dev/_autosummary/jaxopt.IterativeRefinement.html
+++ b/dev/_autosummary/jaxopt.IterativeRefinement.html
@@ -4,7 +4,7 @@
- jaxopt.IterativeRefinement — JAXopt 0.8 documentation
+ jaxopt.IterativeRefinement — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.LBFGS.html b/dev/_autosummary/jaxopt.LBFGS.html
index 290c648..8c7bec6 100644
--- a/dev/_autosummary/jaxopt.LBFGS.html
+++ b/dev/_autosummary/jaxopt.LBFGS.html
@@ -4,7 +4,7 @@
- jaxopt.LBFGS — JAXopt 0.8 documentation
+ jaxopt.LBFGS — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.LBFGSB.html b/dev/_autosummary/jaxopt.LBFGSB.html
index c93d733..be542ff 100644
--- a/dev/_autosummary/jaxopt.LBFGSB.html
+++ b/dev/_autosummary/jaxopt.LBFGSB.html
@@ -4,7 +4,7 @@
- jaxopt.LBFGSB — JAXopt 0.8 documentation
+ jaxopt.LBFGSB — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.LevenbergMarquardt.html b/dev/_autosummary/jaxopt.LevenbergMarquardt.html
index 2f7377e..5761902 100644
--- a/dev/_autosummary/jaxopt.LevenbergMarquardt.html
+++ b/dev/_autosummary/jaxopt.LevenbergMarquardt.html
@@ -4,7 +4,7 @@
- jaxopt.LevenbergMarquardt — JAXopt 0.8 documentation
+ jaxopt.LevenbergMarquardt — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.MirrorDescent.html b/dev/_autosummary/jaxopt.MirrorDescent.html
index 8fb14c7..6880680 100644
--- a/dev/_autosummary/jaxopt.MirrorDescent.html
+++ b/dev/_autosummary/jaxopt.MirrorDescent.html
@@ -4,7 +4,7 @@
- jaxopt.MirrorDescent — JAXopt 0.8 documentation
+ jaxopt.MirrorDescent — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.NonlinearCG.html b/dev/_autosummary/jaxopt.NonlinearCG.html
index 0964f2e..9488f32 100644
--- a/dev/_autosummary/jaxopt.NonlinearCG.html
+++ b/dev/_autosummary/jaxopt.NonlinearCG.html
@@ -4,7 +4,7 @@
- jaxopt.NonlinearCG — JAXopt 0.8 documentation
+ jaxopt.NonlinearCG — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.OSQP.html b/dev/_autosummary/jaxopt.OSQP.html
index a6a8a65..128ce14 100644
--- a/dev/_autosummary/jaxopt.OSQP.html
+++ b/dev/_autosummary/jaxopt.OSQP.html
@@ -4,7 +4,7 @@
- jaxopt.OSQP — JAXopt 0.8 documentation
+ jaxopt.OSQP — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.OptaxSolver.html b/dev/_autosummary/jaxopt.OptaxSolver.html
index 7547ddb..fe75472 100644
--- a/dev/_autosummary/jaxopt.OptaxSolver.html
+++ b/dev/_autosummary/jaxopt.OptaxSolver.html
@@ -4,7 +4,7 @@
- jaxopt.OptaxSolver — JAXopt 0.8 documentation
+ jaxopt.OptaxSolver — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.PolyakSGD.html b/dev/_autosummary/jaxopt.PolyakSGD.html
index ed61d44..7c23dd1 100644
--- a/dev/_autosummary/jaxopt.PolyakSGD.html
+++ b/dev/_autosummary/jaxopt.PolyakSGD.html
@@ -4,7 +4,7 @@
- jaxopt.PolyakSGD — JAXopt 0.8 documentation
+ jaxopt.PolyakSGD — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.ProjectedGradient.html b/dev/_autosummary/jaxopt.ProjectedGradient.html
index 7bcc27d..87e443e 100644
--- a/dev/_autosummary/jaxopt.ProjectedGradient.html
+++ b/dev/_autosummary/jaxopt.ProjectedGradient.html
@@ -4,7 +4,7 @@
- jaxopt.ProjectedGradient — JAXopt 0.8 documentation
+ jaxopt.ProjectedGradient — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.ProximalGradient.html b/dev/_autosummary/jaxopt.ProximalGradient.html
index 9afe419..2fedafd 100644
--- a/dev/_autosummary/jaxopt.ProximalGradient.html
+++ b/dev/_autosummary/jaxopt.ProximalGradient.html
@@ -4,7 +4,7 @@
- jaxopt.ProximalGradient — JAXopt 0.8 documentation
+ jaxopt.ProximalGradient — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.ScipyBoundedMinimize.html b/dev/_autosummary/jaxopt.ScipyBoundedMinimize.html
index b27ad4f..7bd77c4 100644
--- a/dev/_autosummary/jaxopt.ScipyBoundedMinimize.html
+++ b/dev/_autosummary/jaxopt.ScipyBoundedMinimize.html
@@ -4,7 +4,7 @@
- jaxopt.ScipyBoundedMinimize — JAXopt 0.8 documentation
+ jaxopt.ScipyBoundedMinimize — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.ScipyMinimize.html b/dev/_autosummary/jaxopt.ScipyMinimize.html
index 5a5396a..74bf547 100644
--- a/dev/_autosummary/jaxopt.ScipyMinimize.html
+++ b/dev/_autosummary/jaxopt.ScipyMinimize.html
@@ -4,7 +4,7 @@
- jaxopt.ScipyMinimize — JAXopt 0.8 documentation
+ jaxopt.ScipyMinimize — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.ScipyRootFinding.html b/dev/_autosummary/jaxopt.ScipyRootFinding.html
index 16ab837..2ca1f76 100644
--- a/dev/_autosummary/jaxopt.ScipyRootFinding.html
+++ b/dev/_autosummary/jaxopt.ScipyRootFinding.html
@@ -4,7 +4,7 @@
- jaxopt.ScipyRootFinding — JAXopt 0.8 documentation
+ jaxopt.ScipyRootFinding — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.implicit_diff.custom_fixed_point.html b/dev/_autosummary/jaxopt.implicit_diff.custom_fixed_point.html
index 33d0a31..b36b857 100644
--- a/dev/_autosummary/jaxopt.implicit_diff.custom_fixed_point.html
+++ b/dev/_autosummary/jaxopt.implicit_diff.custom_fixed_point.html
@@ -4,7 +4,7 @@
- jaxopt.implicit_diff.custom_fixed_point — JAXopt 0.8 documentation
+ jaxopt.implicit_diff.custom_fixed_point — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.implicit_diff.custom_root.html b/dev/_autosummary/jaxopt.implicit_diff.custom_root.html
index d94590b..e05ea6b 100644
--- a/dev/_autosummary/jaxopt.implicit_diff.custom_root.html
+++ b/dev/_autosummary/jaxopt.implicit_diff.custom_root.html
@@ -4,7 +4,7 @@
- jaxopt.implicit_diff.custom_root — JAXopt 0.8 documentation
+ jaxopt.implicit_diff.custom_root — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.implicit_diff.root_jvp.html b/dev/_autosummary/jaxopt.implicit_diff.root_jvp.html
index 22f0fcd..3be0f8f 100644
--- a/dev/_autosummary/jaxopt.implicit_diff.root_jvp.html
+++ b/dev/_autosummary/jaxopt.implicit_diff.root_jvp.html
@@ -4,7 +4,7 @@
- jaxopt.implicit_diff.root_jvp — JAXopt 0.8 documentation
+ jaxopt.implicit_diff.root_jvp — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.implicit_diff.root_vjp.html b/dev/_autosummary/jaxopt.implicit_diff.root_vjp.html
index be97b5c..760cc6d 100644
--- a/dev/_autosummary/jaxopt.implicit_diff.root_vjp.html
+++ b/dev/_autosummary/jaxopt.implicit_diff.root_vjp.html
@@ -4,7 +4,7 @@
- jaxopt.implicit_diff.root_vjp — JAXopt 0.8 documentation
+ jaxopt.implicit_diff.root_vjp — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.isotonic.isotonic_l2_pav.html b/dev/_autosummary/jaxopt.isotonic.isotonic_l2_pav.html
index bfa68f9..7cc07b4 100644
--- a/dev/_autosummary/jaxopt.isotonic.isotonic_l2_pav.html
+++ b/dev/_autosummary/jaxopt.isotonic.isotonic_l2_pav.html
@@ -4,7 +4,7 @@
- jaxopt.isotonic.isotonic_l2_pav — JAXopt 0.8 documentation
+ jaxopt.isotonic.isotonic_l2_pav — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.linear_solve.solve_bicgstab.html b/dev/_autosummary/jaxopt.linear_solve.solve_bicgstab.html
index af329d8..11c813b 100644
--- a/dev/_autosummary/jaxopt.linear_solve.solve_bicgstab.html
+++ b/dev/_autosummary/jaxopt.linear_solve.solve_bicgstab.html
@@ -4,7 +4,7 @@
- jaxopt.linear_solve.solve_bicgstab — JAXopt 0.8 documentation
+ jaxopt.linear_solve.solve_bicgstab — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.linear_solve.solve_cg.html b/dev/_autosummary/jaxopt.linear_solve.solve_cg.html
index 082729c..44a6e37 100644
--- a/dev/_autosummary/jaxopt.linear_solve.solve_cg.html
+++ b/dev/_autosummary/jaxopt.linear_solve.solve_cg.html
@@ -4,7 +4,7 @@
- jaxopt.linear_solve.solve_cg — JAXopt 0.8 documentation
+ jaxopt.linear_solve.solve_cg — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.linear_solve.solve_cholesky.html b/dev/_autosummary/jaxopt.linear_solve.solve_cholesky.html
index 1f03bae..5f8e8ac 100644
--- a/dev/_autosummary/jaxopt.linear_solve.solve_cholesky.html
+++ b/dev/_autosummary/jaxopt.linear_solve.solve_cholesky.html
@@ -4,7 +4,7 @@
- jaxopt.linear_solve.solve_cholesky — JAXopt 0.8 documentation
+ jaxopt.linear_solve.solve_cholesky — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.linear_solve.solve_gmres.html b/dev/_autosummary/jaxopt.linear_solve.solve_gmres.html
index b0579cf..d6e34be 100644
--- a/dev/_autosummary/jaxopt.linear_solve.solve_gmres.html
+++ b/dev/_autosummary/jaxopt.linear_solve.solve_gmres.html
@@ -4,7 +4,7 @@
- jaxopt.linear_solve.solve_gmres — JAXopt 0.8 documentation
+ jaxopt.linear_solve.solve_gmres — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.linear_solve.solve_lu.html b/dev/_autosummary/jaxopt.linear_solve.solve_lu.html
index 96488fe..3852b56 100644
--- a/dev/_autosummary/jaxopt.linear_solve.solve_lu.html
+++ b/dev/_autosummary/jaxopt.linear_solve.solve_lu.html
@@ -4,7 +4,7 @@
- jaxopt.linear_solve.solve_lu — JAXopt 0.8 documentation
+ jaxopt.linear_solve.solve_lu — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.linear_solve.solve_normal_cg.html b/dev/_autosummary/jaxopt.linear_solve.solve_normal_cg.html
index 7ff405d..bb1ceff 100644
--- a/dev/_autosummary/jaxopt.linear_solve.solve_normal_cg.html
+++ b/dev/_autosummary/jaxopt.linear_solve.solve_normal_cg.html
@@ -4,7 +4,7 @@
- jaxopt.linear_solve.solve_normal_cg — JAXopt 0.8 documentation
+ jaxopt.linear_solve.solve_normal_cg — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.binary_hinge_loss.html b/dev/_autosummary/jaxopt.loss.binary_hinge_loss.html
index 992a059..696e36e 100644
--- a/dev/_autosummary/jaxopt.loss.binary_hinge_loss.html
+++ b/dev/_autosummary/jaxopt.loss.binary_hinge_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.binary_hinge_loss — JAXopt 0.8 documentation
+ jaxopt.loss.binary_hinge_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.binary_logistic_loss.html b/dev/_autosummary/jaxopt.loss.binary_logistic_loss.html
index ad33169..fd117ef 100644
--- a/dev/_autosummary/jaxopt.loss.binary_logistic_loss.html
+++ b/dev/_autosummary/jaxopt.loss.binary_logistic_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.binary_logistic_loss — JAXopt 0.8 documentation
+ jaxopt.loss.binary_logistic_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.binary_perceptron_loss.html b/dev/_autosummary/jaxopt.loss.binary_perceptron_loss.html
index f120f45..f9f6a2d 100644
--- a/dev/_autosummary/jaxopt.loss.binary_perceptron_loss.html
+++ b/dev/_autosummary/jaxopt.loss.binary_perceptron_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.binary_perceptron_loss — JAXopt 0.8 documentation
+ jaxopt.loss.binary_perceptron_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.binary_sparsemax_loss.html b/dev/_autosummary/jaxopt.loss.binary_sparsemax_loss.html
index af1adaa..b2a2715 100644
--- a/dev/_autosummary/jaxopt.loss.binary_sparsemax_loss.html
+++ b/dev/_autosummary/jaxopt.loss.binary_sparsemax_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.binary_sparsemax_loss — JAXopt 0.8 documentation
+ jaxopt.loss.binary_sparsemax_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.huber_loss.html b/dev/_autosummary/jaxopt.loss.huber_loss.html
index 4b712ea..898a4c2 100644
--- a/dev/_autosummary/jaxopt.loss.huber_loss.html
+++ b/dev/_autosummary/jaxopt.loss.huber_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.huber_loss — JAXopt 0.8 documentation
+ jaxopt.loss.huber_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.multiclass_hinge_loss.html b/dev/_autosummary/jaxopt.loss.multiclass_hinge_loss.html
index 887f5d8..e233856 100644
--- a/dev/_autosummary/jaxopt.loss.multiclass_hinge_loss.html
+++ b/dev/_autosummary/jaxopt.loss.multiclass_hinge_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.multiclass_hinge_loss — JAXopt 0.8 documentation
+ jaxopt.loss.multiclass_hinge_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.multiclass_logistic_loss.html b/dev/_autosummary/jaxopt.loss.multiclass_logistic_loss.html
index 6a252b0..e3e691d 100644
--- a/dev/_autosummary/jaxopt.loss.multiclass_logistic_loss.html
+++ b/dev/_autosummary/jaxopt.loss.multiclass_logistic_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.multiclass_logistic_loss — JAXopt 0.8 documentation
+ jaxopt.loss.multiclass_logistic_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.multiclass_perceptron_loss.html b/dev/_autosummary/jaxopt.loss.multiclass_perceptron_loss.html
index fdae2d4..ea8e028 100644
--- a/dev/_autosummary/jaxopt.loss.multiclass_perceptron_loss.html
+++ b/dev/_autosummary/jaxopt.loss.multiclass_perceptron_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.multiclass_perceptron_loss — JAXopt 0.8 documentation
+ jaxopt.loss.multiclass_perceptron_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.multiclass_sparsemax_loss.html b/dev/_autosummary/jaxopt.loss.multiclass_sparsemax_loss.html
index a43d555..83364db 100644
--- a/dev/_autosummary/jaxopt.loss.multiclass_sparsemax_loss.html
+++ b/dev/_autosummary/jaxopt.loss.multiclass_sparsemax_loss.html
@@ -4,7 +4,7 @@
- jaxopt.loss.multiclass_sparsemax_loss — JAXopt 0.8 documentation
+ jaxopt.loss.multiclass_sparsemax_loss — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.sparse_plus.html b/dev/_autosummary/jaxopt.loss.sparse_plus.html
index eba445d..edaae39 100644
--- a/dev/_autosummary/jaxopt.loss.sparse_plus.html
+++ b/dev/_autosummary/jaxopt.loss.sparse_plus.html
@@ -4,7 +4,7 @@
- jaxopt.loss.sparse_plus — JAXopt 0.8 documentation
+ jaxopt.loss.sparse_plus — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.loss.sparse_sigmoid.html b/dev/_autosummary/jaxopt.loss.sparse_sigmoid.html
index 02dc1a6..01b0b79 100644
--- a/dev/_autosummary/jaxopt.loss.sparse_sigmoid.html
+++ b/dev/_autosummary/jaxopt.loss.sparse_sigmoid.html
@@ -4,7 +4,7 @@
- jaxopt.loss.sparse_sigmoid — JAXopt 0.8 documentation
+ jaxopt.loss.sparse_sigmoid — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.binary_logreg.html b/dev/_autosummary/jaxopt.objective.binary_logreg.html
index 37d681d..62b61b0 100644
--- a/dev/_autosummary/jaxopt.objective.binary_logreg.html
+++ b/dev/_autosummary/jaxopt.objective.binary_logreg.html
@@ -4,7 +4,7 @@
- jaxopt.objective.binary_logreg — JAXopt 0.8 documentation
+ jaxopt.objective.binary_logreg — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg.html b/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg.html
index 0a02955..39f6123 100644
--- a/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg.html
+++ b/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg.html
@@ -4,7 +4,7 @@
- jaxopt.objective.l2_multiclass_logreg — JAXopt 0.8 documentation
+ jaxopt.objective.l2_multiclass_logreg — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg_with_intercept.html b/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg_with_intercept.html
index 603f7b5..31d70a7 100644
--- a/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg_with_intercept.html
+++ b/dev/_autosummary/jaxopt.objective.l2_multiclass_logreg_with_intercept.html
@@ -4,7 +4,7 @@
- jaxopt.objective.l2_multiclass_logreg_with_intercept — JAXopt 0.8 documentation
+ jaxopt.objective.l2_multiclass_logreg_with_intercept — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.least_squares.html b/dev/_autosummary/jaxopt.objective.least_squares.html
index daa7c21..82718d2 100644
--- a/dev/_autosummary/jaxopt.objective.least_squares.html
+++ b/dev/_autosummary/jaxopt.objective.least_squares.html
@@ -4,7 +4,7 @@
- jaxopt.objective.least_squares — JAXopt 0.8 documentation
+ jaxopt.objective.least_squares — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.multiclass_linear_svm_dual.html b/dev/_autosummary/jaxopt.objective.multiclass_linear_svm_dual.html
index 84aab92..d11d892 100644
--- a/dev/_autosummary/jaxopt.objective.multiclass_linear_svm_dual.html
+++ b/dev/_autosummary/jaxopt.objective.multiclass_linear_svm_dual.html
@@ -4,7 +4,7 @@
- jaxopt.objective.multiclass_linear_svm_dual — JAXopt 0.8 documentation
+ jaxopt.objective.multiclass_linear_svm_dual — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.multiclass_logreg.html b/dev/_autosummary/jaxopt.objective.multiclass_logreg.html
index 934fd60..dea57c4 100644
--- a/dev/_autosummary/jaxopt.objective.multiclass_logreg.html
+++ b/dev/_autosummary/jaxopt.objective.multiclass_logreg.html
@@ -4,7 +4,7 @@
- jaxopt.objective.multiclass_logreg — JAXopt 0.8 documentation
+ jaxopt.objective.multiclass_logreg — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.multiclass_logreg_with_intercept.html b/dev/_autosummary/jaxopt.objective.multiclass_logreg_with_intercept.html
index d68181e..0f2b0ef 100644
--- a/dev/_autosummary/jaxopt.objective.multiclass_logreg_with_intercept.html
+++ b/dev/_autosummary/jaxopt.objective.multiclass_logreg_with_intercept.html
@@ -4,7 +4,7 @@
- jaxopt.objective.multiclass_logreg_with_intercept — JAXopt 0.8 documentation
+ jaxopt.objective.multiclass_logreg_with_intercept — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.objective.ridge_regression.html b/dev/_autosummary/jaxopt.objective.ridge_regression.html
index 4bf824e..8fa10fb 100644
--- a/dev/_autosummary/jaxopt.objective.ridge_regression.html
+++ b/dev/_autosummary/jaxopt.objective.ridge_regression.html
@@ -4,7 +4,7 @@
- jaxopt.objective.ridge_regression — JAXopt 0.8 documentation
+ jaxopt.objective.ridge_regression — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.perturbations.Gumbel.html b/dev/_autosummary/jaxopt.perturbations.Gumbel.html
index c5113a4..7fd0558 100644
--- a/dev/_autosummary/jaxopt.perturbations.Gumbel.html
+++ b/dev/_autosummary/jaxopt.perturbations.Gumbel.html
@@ -4,7 +4,7 @@
- jaxopt.perturbations.Gumbel — JAXopt 0.8 documentation
+ jaxopt.perturbations.Gumbel — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.perturbations.Normal.html b/dev/_autosummary/jaxopt.perturbations.Normal.html
index 3988134..cec67b8 100644
--- a/dev/_autosummary/jaxopt.perturbations.Normal.html
+++ b/dev/_autosummary/jaxopt.perturbations.Normal.html
@@ -4,7 +4,7 @@
- jaxopt.perturbations.Normal — JAXopt 0.8 documentation
+ jaxopt.perturbations.Normal — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.perturbations.make_perturbed_argmax.html b/dev/_autosummary/jaxopt.perturbations.make_perturbed_argmax.html
index 6282c9d..96cb6ae 100644
--- a/dev/_autosummary/jaxopt.perturbations.make_perturbed_argmax.html
+++ b/dev/_autosummary/jaxopt.perturbations.make_perturbed_argmax.html
@@ -4,7 +4,7 @@
- jaxopt.perturbations.make_perturbed_argmax — JAXopt 0.8 documentation
+ jaxopt.perturbations.make_perturbed_argmax — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.perturbations.make_perturbed_fun.html b/dev/_autosummary/jaxopt.perturbations.make_perturbed_fun.html
index 815964d..199501f 100644
--- a/dev/_autosummary/jaxopt.perturbations.make_perturbed_fun.html
+++ b/dev/_autosummary/jaxopt.perturbations.make_perturbed_fun.html
@@ -4,7 +4,7 @@
- jaxopt.perturbations.make_perturbed_fun — JAXopt 0.8 documentation
+ jaxopt.perturbations.make_perturbed_fun — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.perturbations.make_perturbed_max.html b/dev/_autosummary/jaxopt.perturbations.make_perturbed_max.html
index de652ae..758afe3 100644
--- a/dev/_autosummary/jaxopt.perturbations.make_perturbed_max.html
+++ b/dev/_autosummary/jaxopt.perturbations.make_perturbed_max.html
@@ -4,7 +4,7 @@
- jaxopt.perturbations.make_perturbed_max — JAXopt 0.8 documentation
+ jaxopt.perturbations.make_perturbed_max — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.kl_projection_birkhoff.html b/dev/_autosummary/jaxopt.projection.kl_projection_birkhoff.html
index ac0213a..edd16e3 100644
--- a/dev/_autosummary/jaxopt.projection.kl_projection_birkhoff.html
+++ b/dev/_autosummary/jaxopt.projection.kl_projection_birkhoff.html
@@ -4,7 +4,7 @@
- jaxopt.projection.kl_projection_birkhoff — JAXopt 0.8 documentation
+ jaxopt.projection.kl_projection_birkhoff — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.kl_projection_transport.html b/dev/_autosummary/jaxopt.projection.kl_projection_transport.html
index ecfdf19..64d5bd2 100644
--- a/dev/_autosummary/jaxopt.projection.kl_projection_transport.html
+++ b/dev/_autosummary/jaxopt.projection.kl_projection_transport.html
@@ -4,7 +4,7 @@
- jaxopt.projection.kl_projection_transport — JAXopt 0.8 documentation
+ jaxopt.projection.kl_projection_transport — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_affine_set.html b/dev/_autosummary/jaxopt.projection.projection_affine_set.html
index 1472fd8..ecb71e4 100644
--- a/dev/_autosummary/jaxopt.projection.projection_affine_set.html
+++ b/dev/_autosummary/jaxopt.projection.projection_affine_set.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_affine_set — JAXopt 0.8 documentation
+ jaxopt.projection.projection_affine_set — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_birkhoff.html b/dev/_autosummary/jaxopt.projection.projection_birkhoff.html
index 69c8c3c..17e7e3c 100644
--- a/dev/_autosummary/jaxopt.projection.projection_birkhoff.html
+++ b/dev/_autosummary/jaxopt.projection.projection_birkhoff.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_birkhoff — JAXopt 0.8 documentation
+ jaxopt.projection.projection_birkhoff — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_box.html b/dev/_autosummary/jaxopt.projection.projection_box.html
index 4156bbc..fcb5433 100644
--- a/dev/_autosummary/jaxopt.projection.projection_box.html
+++ b/dev/_autosummary/jaxopt.projection.projection_box.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_box — JAXopt 0.8 documentation
+ jaxopt.projection.projection_box — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_box_section.html b/dev/_autosummary/jaxopt.projection.projection_box_section.html
index f3e2d4e..ed898e3 100644
--- a/dev/_autosummary/jaxopt.projection.projection_box_section.html
+++ b/dev/_autosummary/jaxopt.projection.projection_box_section.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_box_section — JAXopt 0.8 documentation
+ jaxopt.projection.projection_box_section — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_halfspace.html b/dev/_autosummary/jaxopt.projection.projection_halfspace.html
index 2ff2788..f72f6cf 100644
--- a/dev/_autosummary/jaxopt.projection.projection_halfspace.html
+++ b/dev/_autosummary/jaxopt.projection.projection_halfspace.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_halfspace — JAXopt 0.8 documentation
+ jaxopt.projection.projection_halfspace — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_hyperplane.html b/dev/_autosummary/jaxopt.projection.projection_hyperplane.html
index 35a694b..8f45f9c 100644
--- a/dev/_autosummary/jaxopt.projection.projection_hyperplane.html
+++ b/dev/_autosummary/jaxopt.projection.projection_hyperplane.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_hyperplane — JAXopt 0.8 documentation
+ jaxopt.projection.projection_hyperplane — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_l1_ball.html b/dev/_autosummary/jaxopt.projection.projection_l1_ball.html
index 6c04c1d..dd5e738 100644
--- a/dev/_autosummary/jaxopt.projection.projection_l1_ball.html
+++ b/dev/_autosummary/jaxopt.projection.projection_l1_ball.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_l1_ball — JAXopt 0.8 documentation
+ jaxopt.projection.projection_l1_ball — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_l1_sphere.html b/dev/_autosummary/jaxopt.projection.projection_l1_sphere.html
index ac6febf..bb68593 100644
--- a/dev/_autosummary/jaxopt.projection.projection_l1_sphere.html
+++ b/dev/_autosummary/jaxopt.projection.projection_l1_sphere.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_l1_sphere — JAXopt 0.8 documentation
+ jaxopt.projection.projection_l1_sphere — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_l2_ball.html b/dev/_autosummary/jaxopt.projection.projection_l2_ball.html
index 7c8f581..4c97cb4 100644
--- a/dev/_autosummary/jaxopt.projection.projection_l2_ball.html
+++ b/dev/_autosummary/jaxopt.projection.projection_l2_ball.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_l2_ball — JAXopt 0.8 documentation
+ jaxopt.projection.projection_l2_ball — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_l2_sphere.html b/dev/_autosummary/jaxopt.projection.projection_l2_sphere.html
index 665dfa9..adc247f 100644
--- a/dev/_autosummary/jaxopt.projection.projection_l2_sphere.html
+++ b/dev/_autosummary/jaxopt.projection.projection_l2_sphere.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_l2_sphere — JAXopt 0.8 documentation
+ jaxopt.projection.projection_l2_sphere — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_linf_ball.html b/dev/_autosummary/jaxopt.projection.projection_linf_ball.html
index d94121a..1960387 100644
--- a/dev/_autosummary/jaxopt.projection.projection_linf_ball.html
+++ b/dev/_autosummary/jaxopt.projection.projection_linf_ball.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_linf_ball — JAXopt 0.8 documentation
+ jaxopt.projection.projection_linf_ball — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_non_negative.html b/dev/_autosummary/jaxopt.projection.projection_non_negative.html
index 39f6a72..db628fc 100644
--- a/dev/_autosummary/jaxopt.projection.projection_non_negative.html
+++ b/dev/_autosummary/jaxopt.projection.projection_non_negative.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_non_negative — JAXopt 0.8 documentation
+ jaxopt.projection.projection_non_negative — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_polyhedron.html b/dev/_autosummary/jaxopt.projection.projection_polyhedron.html
index dbf170e..7a91072 100644
--- a/dev/_autosummary/jaxopt.projection.projection_polyhedron.html
+++ b/dev/_autosummary/jaxopt.projection.projection_polyhedron.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_polyhedron — JAXopt 0.8 documentation
+ jaxopt.projection.projection_polyhedron — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_simplex.html b/dev/_autosummary/jaxopt.projection.projection_simplex.html
index 3ef03a5..6f701b3 100644
--- a/dev/_autosummary/jaxopt.projection.projection_simplex.html
+++ b/dev/_autosummary/jaxopt.projection.projection_simplex.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_simplex — JAXopt 0.8 documentation
+ jaxopt.projection.projection_simplex — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_sparse_simplex.html b/dev/_autosummary/jaxopt.projection.projection_sparse_simplex.html
index 47d596b..e454732 100644
--- a/dev/_autosummary/jaxopt.projection.projection_sparse_simplex.html
+++ b/dev/_autosummary/jaxopt.projection.projection_sparse_simplex.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_sparse_simplex — JAXopt 0.8 documentation
+ jaxopt.projection.projection_sparse_simplex — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.projection.projection_transport.html b/dev/_autosummary/jaxopt.projection.projection_transport.html
index c4a3892..f27e1a0 100644
--- a/dev/_autosummary/jaxopt.projection.projection_transport.html
+++ b/dev/_autosummary/jaxopt.projection.projection_transport.html
@@ -4,7 +4,7 @@
- jaxopt.projection.projection_transport — JAXopt 0.8 documentation
+ jaxopt.projection.projection_transport — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.make_prox_from_projection.html b/dev/_autosummary/jaxopt.prox.make_prox_from_projection.html
index 1d0738c..0ec5a0f 100644
--- a/dev/_autosummary/jaxopt.prox.make_prox_from_projection.html
+++ b/dev/_autosummary/jaxopt.prox.make_prox_from_projection.html
@@ -4,7 +4,7 @@
- jaxopt.prox.make_prox_from_projection — JAXopt 0.8 documentation
+ jaxopt.prox.make_prox_from_projection — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.prox_elastic_net.html b/dev/_autosummary/jaxopt.prox.prox_elastic_net.html
index f681454..2ddfd90 100644
--- a/dev/_autosummary/jaxopt.prox.prox_elastic_net.html
+++ b/dev/_autosummary/jaxopt.prox.prox_elastic_net.html
@@ -4,7 +4,7 @@
- jaxopt.prox.prox_elastic_net — JAXopt 0.8 documentation
+ jaxopt.prox.prox_elastic_net — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.prox_group_lasso.html b/dev/_autosummary/jaxopt.prox.prox_group_lasso.html
index 90daedc..c8950e6 100644
--- a/dev/_autosummary/jaxopt.prox.prox_group_lasso.html
+++ b/dev/_autosummary/jaxopt.prox.prox_group_lasso.html
@@ -4,7 +4,7 @@
- jaxopt.prox.prox_group_lasso — JAXopt 0.8 documentation
+ jaxopt.prox.prox_group_lasso — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.prox_lasso.html b/dev/_autosummary/jaxopt.prox.prox_lasso.html
index 6069283..fc2fb7e 100644
--- a/dev/_autosummary/jaxopt.prox.prox_lasso.html
+++ b/dev/_autosummary/jaxopt.prox.prox_lasso.html
@@ -4,7 +4,7 @@
- jaxopt.prox.prox_lasso — JAXopt 0.8 documentation
+ jaxopt.prox.prox_lasso — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.prox_non_negative_lasso.html b/dev/_autosummary/jaxopt.prox.prox_non_negative_lasso.html
index 7c94692..d873ea2 100644
--- a/dev/_autosummary/jaxopt.prox.prox_non_negative_lasso.html
+++ b/dev/_autosummary/jaxopt.prox.prox_non_negative_lasso.html
@@ -4,7 +4,7 @@
- jaxopt.prox.prox_non_negative_lasso — JAXopt 0.8 documentation
+ jaxopt.prox.prox_non_negative_lasso — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.prox_non_negative_ridge.html b/dev/_autosummary/jaxopt.prox.prox_non_negative_ridge.html
index f48c837..a8d2ed8 100644
--- a/dev/_autosummary/jaxopt.prox.prox_non_negative_ridge.html
+++ b/dev/_autosummary/jaxopt.prox.prox_non_negative_ridge.html
@@ -4,7 +4,7 @@
- jaxopt.prox.prox_non_negative_ridge — JAXopt 0.8 documentation
+ jaxopt.prox.prox_non_negative_ridge — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.prox_none.html b/dev/_autosummary/jaxopt.prox.prox_none.html
index 416b0e4..600539a 100644
--- a/dev/_autosummary/jaxopt.prox.prox_none.html
+++ b/dev/_autosummary/jaxopt.prox.prox_none.html
@@ -4,7 +4,7 @@
- jaxopt.prox.prox_none — JAXopt 0.8 documentation
+ jaxopt.prox.prox_none — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.prox.prox_ridge.html b/dev/_autosummary/jaxopt.prox.prox_ridge.html
index ec0e0ea..c8f4df0 100644
--- a/dev/_autosummary/jaxopt.prox.prox_ridge.html
+++ b/dev/_autosummary/jaxopt.prox.prox_ridge.html
@@ -4,7 +4,7 @@
- jaxopt.prox.prox_ridge — JAXopt 0.8 documentation
+ jaxopt.prox.prox_ridge — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_add.html b/dev/_autosummary/jaxopt.tree_util.tree_add.html
index 4ee2349..331cad8 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_add.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_add.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_add — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_add — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_add_scalar_mul.html b/dev/_autosummary/jaxopt.tree_util.tree_add_scalar_mul.html
index 54c0ead..8b15d35 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_add_scalar_mul.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_add_scalar_mul.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_add_scalar_mul — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_add_scalar_mul — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_div.html b/dev/_autosummary/jaxopt.tree_util.tree_div.html
index 7c90ecf..9728e74 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_div.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_div.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_div — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_div — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_l2_norm.html b/dev/_autosummary/jaxopt.tree_util.tree_l2_norm.html
index 6659e11..d1b071b 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_l2_norm.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_l2_norm.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_l2_norm — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_l2_norm — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_mul.html b/dev/_autosummary/jaxopt.tree_util.tree_mul.html
index 19358d1..952fb0f 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_mul.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_mul.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_mul — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_mul — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_scalar_mul.html b/dev/_autosummary/jaxopt.tree_util.tree_scalar_mul.html
index 1e35d40..6c61684 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_scalar_mul.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_scalar_mul.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_scalar_mul — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_scalar_mul — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_sub.html b/dev/_autosummary/jaxopt.tree_util.tree_sub.html
index dfc241e..55f5796 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_sub.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_sub.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_sub — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_sub — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_sum.html b/dev/_autosummary/jaxopt.tree_util.tree_sum.html
index 505ac5f..cc85434 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_sum.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_sum.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_sum — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_sum — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_vdot.html b/dev/_autosummary/jaxopt.tree_util.tree_vdot.html
index 240a832..734cd6f 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_vdot.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_vdot.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_vdot — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_vdot — JAXopt 0.8.1 documentation
diff --git a/dev/_autosummary/jaxopt.tree_util.tree_zeros_like.html b/dev/_autosummary/jaxopt.tree_util.tree_zeros_like.html
index 37ab326..2fde881 100644
--- a/dev/_autosummary/jaxopt.tree_util.tree_zeros_like.html
+++ b/dev/_autosummary/jaxopt.tree_util.tree_zeros_like.html
@@ -4,7 +4,7 @@
- jaxopt.tree_util.tree_zeros_like — JAXopt 0.8 documentation
+ jaxopt.tree_util.tree_zeros_like — JAXopt 0.8.1 documentation
diff --git a/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip b/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip
index 407aa24..587a8d2 100644
Binary files a/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip and b/dev/_downloads/07fcc19ba03226cd3d83d4e40ec44385/auto_examples_python.zip differ
diff --git a/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip b/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip
index d12326f..3f9c118 100644
Binary files a/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip and b/dev/_downloads/6f1e7a639e0699d6164445b55e6c116d/auto_examples_jupyter.zip differ
diff --git a/dev/_modules/index.html b/dev/_modules/index.html
index 924651e..7d65dbc 100644
--- a/dev/_modules/index.html
+++ b/dev/_modules/index.html
@@ -3,7 +3,7 @@
- Overview: module code — JAXopt 0.8 documentation
+ Overview: module code — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/anderson.html b/dev/_modules/jaxopt/_src/anderson.html
index 40f973a..b99aa57 100644
--- a/dev/_modules/jaxopt/_src/anderson.html
+++ b/dev/_modules/jaxopt/_src/anderson.html
@@ -3,7 +3,7 @@
- jaxopt._src.anderson — JAXopt 0.8 documentation
+ jaxopt._src.anderson — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/anderson_wrapper.html b/dev/_modules/jaxopt/_src/anderson_wrapper.html
index 4cd9aef..951c28f 100644
--- a/dev/_modules/jaxopt/_src/anderson_wrapper.html
+++ b/dev/_modules/jaxopt/_src/anderson_wrapper.html
@@ -3,7 +3,7 @@
- jaxopt._src.anderson_wrapper — JAXopt 0.8 documentation
+ jaxopt._src.anderson_wrapper — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/armijo_sgd.html b/dev/_modules/jaxopt/_src/armijo_sgd.html
index 45e6f55..51a274e 100644
--- a/dev/_modules/jaxopt/_src/armijo_sgd.html
+++ b/dev/_modules/jaxopt/_src/armijo_sgd.html
@@ -3,7 +3,7 @@
- jaxopt._src.armijo_sgd — JAXopt 0.8 documentation
+ jaxopt._src.armijo_sgd — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/backtracking_linesearch.html b/dev/_modules/jaxopt/_src/backtracking_linesearch.html
index c52babc..7f8d468 100644
--- a/dev/_modules/jaxopt/_src/backtracking_linesearch.html
+++ b/dev/_modules/jaxopt/_src/backtracking_linesearch.html
@@ -3,7 +3,7 @@
- jaxopt._src.backtracking_linesearch — JAXopt 0.8 documentation
+ jaxopt._src.backtracking_linesearch — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/bfgs.html b/dev/_modules/jaxopt/_src/bfgs.html
index ada9191..2395770 100644
--- a/dev/_modules/jaxopt/_src/bfgs.html
+++ b/dev/_modules/jaxopt/_src/bfgs.html
@@ -3,7 +3,7 @@
- jaxopt._src.bfgs — JAXopt 0.8 documentation
+ jaxopt._src.bfgs — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/bisection.html b/dev/_modules/jaxopt/_src/bisection.html
index 10fd218..9fedf61 100644
--- a/dev/_modules/jaxopt/_src/bisection.html
+++ b/dev/_modules/jaxopt/_src/bisection.html
@@ -3,7 +3,7 @@
- jaxopt._src.bisection — JAXopt 0.8 documentation
+ jaxopt._src.bisection — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/block_cd.html b/dev/_modules/jaxopt/_src/block_cd.html
index 54ede63..b774cf0 100644
--- a/dev/_modules/jaxopt/_src/block_cd.html
+++ b/dev/_modules/jaxopt/_src/block_cd.html
@@ -3,7 +3,7 @@
- jaxopt._src.block_cd — JAXopt 0.8 documentation
+ jaxopt._src.block_cd — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/broyden.html b/dev/_modules/jaxopt/_src/broyden.html
index 727017f..9fd3395 100644
--- a/dev/_modules/jaxopt/_src/broyden.html
+++ b/dev/_modules/jaxopt/_src/broyden.html
@@ -3,7 +3,7 @@
- jaxopt._src.broyden — JAXopt 0.8 documentation
+ jaxopt._src.broyden — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/cd_qp.html b/dev/_modules/jaxopt/_src/cd_qp.html
index bd90f2a..86700ec 100644
--- a/dev/_modules/jaxopt/_src/cd_qp.html
+++ b/dev/_modules/jaxopt/_src/cd_qp.html
@@ -3,7 +3,7 @@
- jaxopt._src.cd_qp — JAXopt 0.8 documentation
+ jaxopt._src.cd_qp — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/cvxpy_wrapper.html b/dev/_modules/jaxopt/_src/cvxpy_wrapper.html
index ba52a5b..cb5fba7 100644
--- a/dev/_modules/jaxopt/_src/cvxpy_wrapper.html
+++ b/dev/_modules/jaxopt/_src/cvxpy_wrapper.html
@@ -3,7 +3,7 @@
- jaxopt._src.cvxpy_wrapper — JAXopt 0.8 documentation
+ jaxopt._src.cvxpy_wrapper — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/eq_qp.html b/dev/_modules/jaxopt/_src/eq_qp.html
index 9e9874f..dcf6bdf 100644
--- a/dev/_modules/jaxopt/_src/eq_qp.html
+++ b/dev/_modules/jaxopt/_src/eq_qp.html
@@ -3,7 +3,7 @@
- jaxopt._src.eq_qp — JAXopt 0.8 documentation
+ jaxopt._src.eq_qp — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/fixed_point_iteration.html b/dev/_modules/jaxopt/_src/fixed_point_iteration.html
index 81105f1..727af0c 100644
--- a/dev/_modules/jaxopt/_src/fixed_point_iteration.html
+++ b/dev/_modules/jaxopt/_src/fixed_point_iteration.html
@@ -3,7 +3,7 @@
- jaxopt._src.fixed_point_iteration — JAXopt 0.8 documentation
+ jaxopt._src.fixed_point_iteration — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/gauss_newton.html b/dev/_modules/jaxopt/_src/gauss_newton.html
index edf4b5b..4b72ea1 100644
--- a/dev/_modules/jaxopt/_src/gauss_newton.html
+++ b/dev/_modules/jaxopt/_src/gauss_newton.html
@@ -3,7 +3,7 @@
- jaxopt._src.gauss_newton — JAXopt 0.8 documentation
+ jaxopt._src.gauss_newton — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/gradient_descent.html b/dev/_modules/jaxopt/_src/gradient_descent.html
index d2bcab3..3d716fa 100644
--- a/dev/_modules/jaxopt/_src/gradient_descent.html
+++ b/dev/_modules/jaxopt/_src/gradient_descent.html
@@ -3,7 +3,7 @@
- jaxopt._src.gradient_descent — JAXopt 0.8 documentation
+ jaxopt._src.gradient_descent — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/hager_zhang_linesearch.html b/dev/_modules/jaxopt/_src/hager_zhang_linesearch.html
index 1029b47..81b3e7e 100644
--- a/dev/_modules/jaxopt/_src/hager_zhang_linesearch.html
+++ b/dev/_modules/jaxopt/_src/hager_zhang_linesearch.html
@@ -3,7 +3,7 @@
- jaxopt._src.hager_zhang_linesearch — JAXopt 0.8 documentation
+ jaxopt._src.hager_zhang_linesearch — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/implicit_diff.html b/dev/_modules/jaxopt/_src/implicit_diff.html
index d314062..2975598 100644
--- a/dev/_modules/jaxopt/_src/implicit_diff.html
+++ b/dev/_modules/jaxopt/_src/implicit_diff.html
@@ -3,7 +3,7 @@
- jaxopt._src.implicit_diff — JAXopt 0.8 documentation
+ jaxopt._src.implicit_diff — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/isotonic.html b/dev/_modules/jaxopt/_src/isotonic.html
index 9400f35..341b468 100644
--- a/dev/_modules/jaxopt/_src/isotonic.html
+++ b/dev/_modules/jaxopt/_src/isotonic.html
@@ -3,7 +3,7 @@
- jaxopt._src.isotonic — JAXopt 0.8 documentation
+ jaxopt._src.isotonic — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/iterative_refinement.html b/dev/_modules/jaxopt/_src/iterative_refinement.html
index 62e87d1..4129cca 100644
--- a/dev/_modules/jaxopt/_src/iterative_refinement.html
+++ b/dev/_modules/jaxopt/_src/iterative_refinement.html
@@ -3,7 +3,7 @@
- jaxopt._src.iterative_refinement — JAXopt 0.8 documentation
+ jaxopt._src.iterative_refinement — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/lbfgs.html b/dev/_modules/jaxopt/_src/lbfgs.html
index 0a107f3..292ae42 100644
--- a/dev/_modules/jaxopt/_src/lbfgs.html
+++ b/dev/_modules/jaxopt/_src/lbfgs.html
@@ -3,7 +3,7 @@
- jaxopt._src.lbfgs — JAXopt 0.8 documentation
+ jaxopt._src.lbfgs — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/lbfgsb.html b/dev/_modules/jaxopt/_src/lbfgsb.html
index a73ed34..a672daa 100644
--- a/dev/_modules/jaxopt/_src/lbfgsb.html
+++ b/dev/_modules/jaxopt/_src/lbfgsb.html
@@ -3,7 +3,7 @@
- jaxopt._src.lbfgsb — JAXopt 0.8 documentation
+ jaxopt._src.lbfgsb — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/levenberg_marquardt.html b/dev/_modules/jaxopt/_src/levenberg_marquardt.html
index 0420bc9..addfa79 100644
--- a/dev/_modules/jaxopt/_src/levenberg_marquardt.html
+++ b/dev/_modules/jaxopt/_src/levenberg_marquardt.html
@@ -3,7 +3,7 @@
- jaxopt._src.levenberg_marquardt — JAXopt 0.8 documentation
+ jaxopt._src.levenberg_marquardt — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/linear_solve.html b/dev/_modules/jaxopt/_src/linear_solve.html
index b6f136c..df4347e 100644
--- a/dev/_modules/jaxopt/_src/linear_solve.html
+++ b/dev/_modules/jaxopt/_src/linear_solve.html
@@ -3,7 +3,7 @@
- jaxopt._src.linear_solve — JAXopt 0.8 documentation
+ jaxopt._src.linear_solve — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/loss.html b/dev/_modules/jaxopt/_src/loss.html
index e85e785..c098317 100644
--- a/dev/_modules/jaxopt/_src/loss.html
+++ b/dev/_modules/jaxopt/_src/loss.html
@@ -3,7 +3,7 @@
- jaxopt._src.loss — JAXopt 0.8 documentation
+ jaxopt._src.loss — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/mirror_descent.html b/dev/_modules/jaxopt/_src/mirror_descent.html
index 82bf2f8..599d7f7 100644
--- a/dev/_modules/jaxopt/_src/mirror_descent.html
+++ b/dev/_modules/jaxopt/_src/mirror_descent.html
@@ -3,7 +3,7 @@
- jaxopt._src.mirror_descent — JAXopt 0.8 documentation
+ jaxopt._src.mirror_descent — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/nonlinear_cg.html b/dev/_modules/jaxopt/_src/nonlinear_cg.html
index 49ef312..3819d42 100644
--- a/dev/_modules/jaxopt/_src/nonlinear_cg.html
+++ b/dev/_modules/jaxopt/_src/nonlinear_cg.html
@@ -3,7 +3,7 @@
- jaxopt._src.nonlinear_cg — JAXopt 0.8 documentation
+ jaxopt._src.nonlinear_cg — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/objective.html b/dev/_modules/jaxopt/_src/objective.html
index b80caec..3209219 100644
--- a/dev/_modules/jaxopt/_src/objective.html
+++ b/dev/_modules/jaxopt/_src/objective.html
@@ -3,7 +3,7 @@
- jaxopt._src.objective — JAXopt 0.8 documentation
+ jaxopt._src.objective — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/optax_wrapper.html b/dev/_modules/jaxopt/_src/optax_wrapper.html
index 66e1c33..42055ea 100644
--- a/dev/_modules/jaxopt/_src/optax_wrapper.html
+++ b/dev/_modules/jaxopt/_src/optax_wrapper.html
@@ -3,7 +3,7 @@
- jaxopt._src.optax_wrapper — JAXopt 0.8 documentation
+ jaxopt._src.optax_wrapper — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/osqp.html b/dev/_modules/jaxopt/_src/osqp.html
index feecb02..50024af 100644
--- a/dev/_modules/jaxopt/_src/osqp.html
+++ b/dev/_modules/jaxopt/_src/osqp.html
@@ -3,7 +3,7 @@
- jaxopt._src.osqp — JAXopt 0.8 documentation
+ jaxopt._src.osqp — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/perturbations.html b/dev/_modules/jaxopt/_src/perturbations.html
index e1a6db0..50c2282 100644
--- a/dev/_modules/jaxopt/_src/perturbations.html
+++ b/dev/_modules/jaxopt/_src/perturbations.html
@@ -3,7 +3,7 @@
- jaxopt._src.perturbations — JAXopt 0.8 documentation
+ jaxopt._src.perturbations — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/polyak_sgd.html b/dev/_modules/jaxopt/_src/polyak_sgd.html
index 5944f79..14c3728 100644
--- a/dev/_modules/jaxopt/_src/polyak_sgd.html
+++ b/dev/_modules/jaxopt/_src/polyak_sgd.html
@@ -3,7 +3,7 @@
- jaxopt._src.polyak_sgd — JAXopt 0.8 documentation
+ jaxopt._src.polyak_sgd — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/projected_gradient.html b/dev/_modules/jaxopt/_src/projected_gradient.html
index 57b8969..b5b8028 100644
--- a/dev/_modules/jaxopt/_src/projected_gradient.html
+++ b/dev/_modules/jaxopt/_src/projected_gradient.html
@@ -3,7 +3,7 @@
- jaxopt._src.projected_gradient — JAXopt 0.8 documentation
+ jaxopt._src.projected_gradient — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/projection.html b/dev/_modules/jaxopt/_src/projection.html
index 5e00d1b..afbf20e 100644
--- a/dev/_modules/jaxopt/_src/projection.html
+++ b/dev/_modules/jaxopt/_src/projection.html
@@ -3,7 +3,7 @@
- jaxopt._src.projection — JAXopt 0.8 documentation
+ jaxopt._src.projection — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/prox.html b/dev/_modules/jaxopt/_src/prox.html
index c6c82dd..4bcd864 100644
--- a/dev/_modules/jaxopt/_src/prox.html
+++ b/dev/_modules/jaxopt/_src/prox.html
@@ -3,7 +3,7 @@
- jaxopt._src.prox — JAXopt 0.8 documentation
+ jaxopt._src.prox — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/proximal_gradient.html b/dev/_modules/jaxopt/_src/proximal_gradient.html
index 412c25c..3680202 100644
--- a/dev/_modules/jaxopt/_src/proximal_gradient.html
+++ b/dev/_modules/jaxopt/_src/proximal_gradient.html
@@ -3,7 +3,7 @@
- jaxopt._src.proximal_gradient — JAXopt 0.8 documentation
+ jaxopt._src.proximal_gradient — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/scipy_wrappers.html b/dev/_modules/jaxopt/_src/scipy_wrappers.html
index 59c71c2..44f6370 100644
--- a/dev/_modules/jaxopt/_src/scipy_wrappers.html
+++ b/dev/_modules/jaxopt/_src/scipy_wrappers.html
@@ -3,7 +3,7 @@
- jaxopt._src.scipy_wrappers — JAXopt 0.8 documentation
+ jaxopt._src.scipy_wrappers — JAXopt 0.8.1 documentation
diff --git a/dev/_modules/jaxopt/_src/tree_util.html b/dev/_modules/jaxopt/_src/tree_util.html
index 6447ad7..11cfa0d 100644
--- a/dev/_modules/jaxopt/_src/tree_util.html
+++ b/dev/_modules/jaxopt/_src/tree_util.html
@@ -3,7 +3,7 @@
- jaxopt._src.tree_util — JAXopt 0.8 documentation
+ jaxopt._src.tree_util — JAXopt 0.8.1 documentation
diff --git a/dev/_sources/auto_examples/deep_learning/plot_sgd_solvers.rst.txt b/dev/_sources/auto_examples/deep_learning/plot_sgd_solvers.rst.txt
index 1379a33..c091b04 100644
--- a/dev/_sources/auto_examples/deep_learning/plot_sgd_solvers.rst.txt
+++ b/dev/_sources/auto_examples/deep_learning/plot_sgd_solvers.rst.txt
@@ -82,199 +82,212 @@ learning rate.
Extraction completed...: 0 file [00:00, ? file/s]
Dl Completed...: 0%| | 0/4 [00:00, ? url/s]
Dl Size...: 0%| | 0/29 [00:00, ? MiB/s]
-
Extraction completed...: 0 file [00:00, ? file/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
+
Extraction completed...: 0 file [00:00, ? file/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
Dl Size...: 0%| | 0/29 [00:00, ? MiB/s]
-
Extraction completed...: 0 file [00:00, ? file/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
+
Extraction completed...: 0 file [00:00, ? file/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
Dl Size...: 0%| | 0/29 [00:00, ? MiB/s]
Extraction completed...: 0%| | 0/1 [00:00, ? file/s]
-
Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 3.61 file/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
+
Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 2.44 file/s]
Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
Dl Size...: 0%| | 0/29 [00:00, ? MiB/s]
-
Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+
Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 2.44 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
Dl Size...: 0%| | 0/29 [00:00, ? MiB/s]
-
Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+
Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 2.44 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
Dl Size...: 0%| | 0/29 [00:00, ? MiB/s]
-
Extraction completed...: 50%|█████ | 1/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+
Extraction completed...: 50%|█████ | 1/2 [00:00<00:00, 2.44 file/s]
+
+
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.85 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
Dl Size...: 0%| | 0/29 [00:00, ? MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-
Dl Size...: 3%|▎ | 1/29 [00:00<00:22, 1.26 MiB/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-
Dl Size...: 3%|▎ | 1/29 [00:00<00:22, 1.26 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.85 file/s]
+
Dl Size...: 3%|▎ | 1/29 [00:01<00:37, 1.35s/ MiB]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 3%|▎ | 1/29 [00:01<00:37, 1.35s/ MiB]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-
Dl Size...: 7%|▋ | 2/29 [00:00<00:21, 1.26 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 7%|▋ | 2/29 [00:01<00:36, 1.35s/ MiB]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-
Dl Size...: 10%|█ | 3/29 [00:00<00:20, 1.26 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+
Dl Size...: 10%|█ | 3/29 [00:01<00:10, 2.46 MiB/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 10%|█ | 3/29 [00:01<00:10, 2.46 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-
Dl Size...: 14%|█▍ | 4/29 [00:00<00:19, 1.26 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 14%|█▍ | 4/29 [00:01<00:10, 2.46 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-
Dl Size...: 17%|█▋ | 5/29 [00:00<00:03, 7.08 MiB/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-
Dl Size...: 17%|█▋ | 5/29 [00:00<00:03, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 17%|█▋ | 5/29 [00:01<00:09, 2.46 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-
Dl Size...: 21%|██ | 6/29 [00:00<00:03, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 21%|██ | 6/29 [00:01<00:09, 2.46 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-
Dl Size...: 24%|██▍ | 7/29 [00:00<00:03, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+
Dl Size...: 24%|██▍ | 7/29 [00:01<00:03, 6.34 MiB/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 24%|██▍ | 7/29 [00:01<00:03, 6.34 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
-
Dl Size...: 24%|██▍ | 7/29 [00:00<00:03, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 28%|██▊ | 8/29 [00:01<00:03, 6.34 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
-
Dl Size...: 28%|██▊ | 8/29 [00:00<00:02, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
Dl Size...: 31%|███ | 9/29 [00:01<00:03, 6.34 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
-
Dl Size...: 31%|███ | 9/29 [00:00<00:02, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 31%|███ | 9/29 [00:01<00:03, 6.34 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
-
Dl Size...: 34%|███▍ | 10/29 [00:00<00:02, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 6.34 MiB/s]
-
Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 6.34 MiB/s]
-
Extraction completed...: 67%|██████▋ | 2/3 [00:01<00:00, 3.61 file/s]
+
Extraction completed...: 67%|██████▋ | 2/3 [00:01<00:00, 3.85 file/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 7.08 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 6.34 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-
Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 14.68 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 14.68 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
+
Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 9.96 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 9.96 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 41%|████▏ | 12/29 [00:01<00:01, 14.68 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 41%|████▏ | 12/29 [00:01<00:01, 9.96 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 45%|████▍ | 13/29 [00:01<00:01, 14.68 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
+
Dl Size...: 45%|████▍ | 13/29 [00:01<00:01, 10.66 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 45%|████▍ | 13/29 [00:01<00:01, 10.66 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 48%|████▊ | 14/29 [00:01<00:01, 14.68 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 48%|████▊ | 14/29 [00:01<00:01, 10.66 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-
Dl Size...: 52%|█████▏ | 15/29 [00:01<00:00, 19.13 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 52%|█████▏ | 15/29 [00:01<00:00, 19.13 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+
Dl Size...: 52%|█████▏ | 15/29 [00:01<00:01, 10.66 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 55%|█████▌ | 16/29 [00:01<00:00, 19.13 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
+
Dl Size...: 55%|█████▌ | 16/29 [00:02<00:01, 12.58 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 55%|█████▌ | 16/29 [00:02<00:01, 12.58 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 59%|█████▊ | 17/29 [00:01<00:00, 19.13 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 59%|█████▊ | 17/29 [00:02<00:00, 12.58 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 62%|██████▏ | 18/29 [00:01<00:00, 19.13 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+
Dl Size...: 62%|██████▏ | 18/29 [00:02<00:00, 12.72 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 62%|██████▏ | 18/29 [00:02<00:00, 12.72 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-
Dl Size...: 66%|██████▌ | 19/29 [00:01<00:00, 20.41 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 66%|██████▌ | 19/29 [00:01<00:00, 20.41 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 66%|██████▌ | 19/29 [00:02<00:00, 12.72 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 69%|██████▉ | 20/29 [00:01<00:00, 20.41 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 69%|██████▉ | 20/29 [00:02<00:00, 12.72 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 72%|███████▏ | 21/29 [00:01<00:00, 20.41 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+
Dl Size...: 72%|███████▏ | 21/29 [00:02<00:00, 14.21 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 72%|███████▏ | 21/29 [00:02<00:00, 14.21 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 76%|███████▌ | 22/29 [00:01<00:00, 20.41 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 76%|███████▌ | 22/29 [00:02<00:00, 14.21 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-
Dl Size...: 79%|███████▉ | 23/29 [00:01<00:00, 21.60 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 79%|███████▉ | 23/29 [00:01<00:00, 21.60 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+
Dl Size...: 79%|███████▉ | 23/29 [00:02<00:00, 14.07 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 79%|███████▉ | 23/29 [00:02<00:00, 14.07 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 83%|████████▎ | 24/29 [00:01<00:00, 21.60 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 83%|████████▎ | 24/29 [00:02<00:00, 14.07 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 86%|████████▌ | 25/29 [00:01<00:00, 21.60 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+
Dl Size...: 86%|████████▌ | 25/29 [00:02<00:00, 13.71 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 86%|████████▌ | 25/29 [00:02<00:00, 13.71 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 90%|████████▉ | 26/29 [00:01<00:00, 21.60 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 90%|████████▉ | 26/29 [00:02<00:00, 13.71 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-
Dl Size...: 93%|█████████▎| 27/29 [00:01<00:00, 25.00 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 93%|█████████▎| 27/29 [00:01<00:00, 25.00 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 93%|█████████▎| 27/29 [00:02<00:00, 13.71 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 97%|█████████▋| 28/29 [00:01<00:00, 25.00 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+
Dl Size...: 97%|█████████▋| 28/29 [00:02<00:00, 15.21 MiB/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 97%|█████████▋| 28/29 [00:02<00:00, 15.21 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-
Dl Size...: 100%|██████████| 29/29 [00:01<00:00, 25.00 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+
Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 15.21 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 100%|██████████| 4/4 [00:01<00:00, 2.13 url/s]
Dl Completed...: 100%|██████████| 4/4 [00:01<00:00, 2.13 url/s]
-
Dl Size...: 100%|██████████| 29/29 [00:01<00:00, 25.00 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 1.13 url/s]
Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 1.13 url/s]
+
Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 15.21 MiB/s]
-
Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 2.13 url/s]
-
Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 25.00 MiB/s]
+
Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
Dl Completed...: 100%|██████████| 4/4 [00:03<00:00, 1.13 url/s]
+
Dl Size...: 100%|██████████| 29/29 [00:03<00:00, 15.21 MiB/s]
-
Extraction completed...: 75%|███████▌ | 3/4 [00:02<00:00, 2.79 file/s]
+
Extraction completed...: 75%|███████▌ | 3/4 [00:03<00:00, 1.46 file/s]
-
Extraction completed...: 100%|██████████| 4/4 [00:02<00:00, 1.70 file/s]
Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 2.13 url/s]
-
Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 25.00 MiB/s]
+
Extraction completed...: 100%|██████████| 4/4 [00:03<00:00, 1.06s/ file]
Dl Completed...: 100%|██████████| 4/4 [00:03<00:00, 1.13 url/s]
+
Dl Size...: 100%|██████████| 29/29 [00:03<00:00, 15.21 MiB/s]
-
Extraction completed...: 100%|██████████| 4/4 [00:02<00:00, 1.70 file/s]
Extraction completed...: 100%|██████████| 4/4 [00:02<00:00, 1.93 file/s]
-
Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 14.00 MiB/s]
-
Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 1.93 url/s]
+
Extraction completed...: 100%|██████████| 4/4 [00:03<00:00, 1.06s/ file]
Extraction completed...: 100%|██████████| 4/4 [00:03<00:00, 1.18 file/s]
+
Dl Size...: 100%|██████████| 29/29 [00:03<00:00, 8.56 MiB/s]
+
Dl Completed...: 100%|██████████| 4/4 [00:03<00:00, 1.18 url/s]
Generating splits...: 0%| | 0/2 [00:00, ? splits/s]
Generating train examples...: 0%| | 0/60000 [00:00, ? examples/s]
-
Generating train examples...: 3%|▎ | 1690/60000 [00:01<00:34, 1689.98 examples/s]
-
Generating train examples...: 6%|▌ | 3567/60000 [00:02<00:31, 1799.89 examples/s]
-
Generating train examples...: 9%|▉ | 5437/60000 [00:03<00:29, 1831.69 examples/s]
-
Generating train examples...: 12%|█▏ | 7308/60000 [00:04<00:28, 1847.19 examples/s]
-
Generating train examples...: 15%|█▌ | 9180/60000 [00:05<00:27, 1856.03 examples/s]
-
Generating train examples...: 18%|█▊ | 11053/60000 [00:06<00:26, 1861.67 examples/s]
-
Generating train examples...: 22%|██▏ | 12924/60000 [00:07<00:25, 1864.54 examples/s]
-
Generating train examples...: 25%|██▍ | 14798/60000 [00:08<00:24, 1867.37 examples/s]
-
Generating train examples...: 28%|██▊ | 16671/60000 [00:09<00:23, 1869.05 examples/s]
-
Generating train examples...: 31%|███ | 18542/60000 [00:10<00:22, 1869.37 examples/s]
-
Generating train examples...: 34%|███▍ | 20415/60000 [00:11<00:21, 1870.20 examples/s]
-
Generating train examples...: 37%|███▋ | 22290/60000 [00:12<00:20, 1871.52 examples/s]
-
Generating train examples...: 40%|████ | 24162/60000 [00:13<00:19, 1865.31 examples/s]
-
Generating train examples...: 43%|████▎ | 26033/60000 [00:14<00:18, 1866.98 examples/s]
-
Generating train examples...: 47%|████▋ | 27910/60000 [00:15<00:17, 1869.86 examples/s]
-
Generating train examples...: 50%|████▉ | 29788/60000 [00:16<00:16, 1872.02 examples/s]
-
Generating train examples...: 53%|█████▎ | 31668/60000 [00:17<00:15, 1874.25 examples/s]
-
Generating train examples...: 56%|█████▌ | 33544/60000 [00:18<00:14, 1874.63 examples/s]
-
Generating train examples...: 59%|█████▉ | 35421/60000 [00:19<00:13, 1875.26 examples/s]
-
Generating train examples...: 62%|██████▏ | 37297/60000 [00:20<00:12, 1875.43 examples/s]
-
Generating train examples...: 65%|██████▌ | 39175/60000 [00:21<00:11, 1876.16 examples/s]
-
Generating train examples...: 68%|██████▊ | 41055/60000 [00:22<00:10, 1877.14 examples/s]
-
Generating train examples...: 72%|███████▏ | 42935/60000 [00:23<00:09, 1877.82 examples/s]
-
Generating train examples...: 75%|███████▍ | 44813/60000 [00:24<00:08, 1877.17 examples/s]
-
Generating train examples...: 78%|███████▊ | 46691/60000 [00:25<00:07, 1876.61 examples/s]
-
Generating train examples...: 81%|████████ | 48568/60000 [00:26<00:06, 1875.08 examples/s]
-
Generating train examples...: 84%|████████▍ | 50449/60000 [00:27<00:05, 1876.75 examples/s]
-
Generating train examples...: 87%|████████▋ | 52326/60000 [00:28<00:04, 1875.66 examples/s]
-
Generating train examples...: 90%|█████████ | 54205/60000 [00:29<00:03, 1876.52 examples/s]
-
Generating train examples...: 93%|█████████▎| 56082/60000 [00:30<00:02, 1876.61 examples/s]
-
Generating train examples...: 97%|█████████▋| 57960/60000 [00:31<00:01, 1876.76 examples/s]
-
Generating train examples...: 100%|█████████▉| 59837/60000 [00:32<00:00, 1876.49 examples/s]
+
Generating train examples...: 2%|▏ | 1285/60000 [00:01<00:45, 1284.71 examples/s]
+
Generating train examples...: 5%|▍ | 2881/60000 [00:02<00:38, 1467.71 examples/s]
+
Generating train examples...: 7%|▋ | 4488/60000 [00:03<00:36, 1531.03 examples/s]
+
Generating train examples...: 10%|█ | 6094/60000 [00:04<00:34, 1560.59 examples/s]
+
Generating train examples...: 13%|█▎ | 7706/60000 [00:05<00:33, 1579.07 examples/s]
+
Generating train examples...: 16%|█▌ | 9312/60000 [00:06<00:31, 1588.05 examples/s]
+
Generating train examples...: 18%|█▊ | 10904/60000 [00:07<00:30, 1589.29 examples/s]
+
Generating train examples...: 21%|██ | 12511/60000 [00:08<00:29, 1594.64 examples/s]
+
Generating train examples...: 24%|██▎ | 14120/60000 [00:09<00:28, 1598.92 examples/s]
+
Generating train examples...: 26%|██▌ | 15731/60000 [00:10<00:27, 1602.37 examples/s]
+
Generating train examples...: 29%|██▉ | 17341/60000 [00:11<00:26, 1604.65 examples/s]
+
Generating train examples...: 32%|███▏ | 18946/60000 [00:12<00:25, 1602.91 examples/s]
+
Generating train examples...: 34%|███▍ | 20552/60000 [00:13<00:24, 1603.74 examples/s]
+
Generating train examples...: 37%|███▋ | 22160/60000 [00:14<00:23, 1605.01 examples/s]
+
Generating train examples...: 40%|███▉ | 23769/60000 [00:15<00:22, 1605.96 examples/s]
+
Generating train examples...: 42%|████▏ | 25375/60000 [00:16<00:21, 1602.23 examples/s]
+
Generating train examples...: 45%|████▍ | 26983/60000 [00:17<00:20, 1603.86 examples/s]
+
Generating train examples...: 48%|████▊ | 28590/60000 [00:18<00:19, 1604.50 examples/s]
+
Generating train examples...: 50%|█████ | 30196/60000 [00:19<00:18, 1604.81 examples/s]
+
Generating train examples...: 53%|█████▎ | 31801/60000 [00:20<00:17, 1604.09 examples/s]
+
Generating train examples...: 56%|█████▌ | 33407/60000 [00:21<00:16, 1604.40 examples/s]
+
Generating train examples...: 58%|█████▊ | 35014/60000 [00:22<00:15, 1605.06 examples/s]
+
Generating train examples...: 61%|██████ | 36620/60000 [00:23<00:14, 1604.24 examples/s]
+
Generating train examples...: 64%|██████▎ | 38225/60000 [00:24<00:13, 1604.44 examples/s]
+
Generating train examples...: 66%|██████▋ | 39830/60000 [00:25<00:12, 1604.24 examples/s]
+
Generating train examples...: 69%|██████▉ | 41435/60000 [00:26<00:11, 1603.83 examples/s]
+
Generating train examples...: 72%|███████▏ | 43043/60000 [00:27<00:10, 1604.79 examples/s]
+
Generating train examples...: 74%|███████▍ | 44648/60000 [00:28<00:09, 1603.42 examples/s]
+
Generating train examples...: 77%|███████▋ | 46252/60000 [00:29<00:08, 1603.22 examples/s]
+
Generating train examples...: 80%|███████▉ | 47857/60000 [00:30<00:07, 1603.54 examples/s]
+
Generating train examples...: 82%|████████▏ | 49461/60000 [00:31<00:06, 1599.84 examples/s]
+
Generating train examples...: 85%|████████▌ | 51061/60000 [00:32<00:05, 1599.78 examples/s]
+
Generating train examples...: 88%|████████▊ | 52661/60000 [00:33<00:04, 1597.94 examples/s]
+
Generating train examples...: 90%|█████████ | 54263/60000 [00:34<00:03, 1599.08 examples/s]
+
Generating train examples...: 93%|█████████▎| 55863/60000 [00:35<00:02, 1599.19 examples/s]
+
Generating train examples...: 96%|█████████▌| 57464/60000 [00:36<00:01, 1599.43 examples/s]
+
Generating train examples...: 98%|█████████▊| 59064/60000 [00:37<00:00, 1598.35 examples/s]
-
Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteD287AU/fashion_mnist-train.tfrecord*...: 0%| | 0/60000 [00:00, ? examples/s]
-
Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteD287AU/fashion_mnist-train.tfrecord*...: 46%|████▋ | 27864/60000 [00:00<00:00, 278617.68 examples/s]
-
Generating splits...: 50%|█████ | 1/2 [00:32<00:32, 32.31s/ splits]
+
Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-train.tfrecord*...: 0%| | 0/60000 [00:00, ? examples/s]
+
Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-train.tfrecord*...: 30%|███ | 18179/60000 [00:00<00:00, 181777.17 examples/s]
+
Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-train.tfrecord*...: 93%|█████████▎| 55686/60000 [00:00<00:00, 295461.61 examples/s]
+
Generating splits...: 50%|█████ | 1/2 [00:37<00:37, 37.87s/ splits]
Generating test examples...: 0%| | 0/10000 [00:00, ? examples/s]
-
Generating test examples...: 19%|█▊ | 1860/10000 [00:01<00:04, 1859.89 examples/s]
-
Generating test examples...: 37%|███▋ | 3747/10000 [00:02<00:03, 1875.63 examples/s]
-
Generating test examples...: 56%|█████▋ | 5635/10000 [00:03<00:02, 1880.82 examples/s]
-
Generating test examples...: 75%|███████▌ | 7523/10000 [00:04<00:01, 1883.31 examples/s]
-
Generating test examples...: 94%|█████████▍| 9411/10000 [00:05<00:00, 1884.99 examples/s]
+
Generating test examples...: 16%|█▌ | 1580/10000 [00:01<00:05, 1579.80 examples/s]
+
Generating test examples...: 32%|███▏ | 3194/10000 [00:02<00:04, 1599.55 examples/s]
+
Generating test examples...: 48%|████▊ | 4804/10000 [00:03<00:03, 1604.15 examples/s]
+
Generating test examples...: 64%|██████▍ | 6418/10000 [00:04<00:02, 1607.73 examples/s]
+
Generating test examples...: 80%|████████ | 8031/10000 [00:05<00:01, 1609.37 examples/s]
+
Generating test examples...: 96%|█████████▋| 9645/10000 [00:06<00:00, 1610.82 examples/s]
-
Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteD287AU/fashion_mnist-test.tfrecord*...: 0%| | 0/10000 [00:00, ? examples/s]
-
Generating splits...: 100%|██████████| 2/2 [00:37<00:00, 16.45s/ splits]
Dataset fashion_mnist downloaded and prepared to /home/runner/tensorflow_datasets/fashion_mnist/3.0.1. Subsequent calls will reuse this data.
+
Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-test.tfrecord*...: 0%| | 0/10000 [00:00, ? examples/s]
+
Generating splits...: 100%|██████████| 2/2 [00:44<00:00, 19.28s/ splits]
Dataset fashion_mnist downloaded and prepared to /home/runner/tensorflow_datasets/fashion_mnist/3.0.1. Subsequent calls will reuse this data.
[Polyak SGD] ( 0), error=0.272, loss=2.301, stepsize=1.000000
[Polyak SGD] ( 20), error=1.887, loss=1.476, stepsize=0.414197
[Polyak SGD] ( 40), error=2.054, loss=1.342, stepsize=0.317994
[Polyak SGD] ( 60), error=1.689, loss=0.831, stepsize=0.291400
[Polyak SGD] ( 80), error=1.325, loss=1.168, stepsize=0.665042
[Polyak SGD] ( 100), error=1.382, loss=0.916, stepsize=0.479756
- 101 iterations of Polyak SGD took 15.03s
+ 101 iterations of Polyak SGD took 19.83s
[Armijo SGD] ( 0), error=0.227, loss=2.313, stepsize=1.000000
[Armijo SGD] ( 20), error=2.221, loss=1.331, stepsize=0.065971
@@ -282,7 +295,7 @@ learning rate.
[Armijo SGD] ( 60), error=3.499, loss=1.059, stepsize=0.027022
[Armijo SGD] ( 80), error=1.456, loss=0.863, stepsize=0.021617
[Armijo SGD] ( 100), error=1.689, loss=0.819, stepsize=0.021617
- 101 iterations of Armijo SGD took 17.76s
+ 101 iterations of Armijo SGD took 23.17s
[RMSProp] ( 0), error=0.223, loss=2.299
[RMSProp] ( 20), error=0.869, loss=1.913
@@ -290,7 +303,7 @@ learning rate.
[RMSProp] ( 60), error=1.115, loss=1.211
[RMSProp] ( 80), error=1.627, loss=1.088
[RMSProp] ( 100), error=1.581, loss=1.002
- 101 iterations of RMSProp took 13.94s
+ 101 iterations of RMSProp took 17.58s
[SGD] ( 0), error=0.190, loss=2.306
[SGD] ( 20), error=0.384, loss=2.236
@@ -298,7 +311,7 @@ learning rate.
[SGD] ( 60), error=1.222, loss=1.092
[SGD] ( 80), error=1.637, loss=1.146
[SGD] ( 100), error=4.393, loss=1.671
- 101 iterations of SGD took 13.78s
+ 101 iterations of SGD took 17.34s
@@ -492,7 +505,7 @@ learning rate.
.. rst-class:: sphx-glr-timing
- **Total running time of the script:** (1 minutes 44.325 seconds)
+ **Total running time of the script:** (2 minutes 33.975 seconds)
.. _sphx_glr_download_auto_examples_deep_learning_plot_sgd_solvers.py:
diff --git a/dev/_sources/auto_examples/deep_learning/sg_execution_times.rst.txt b/dev/_sources/auto_examples/deep_learning/sg_execution_times.rst.txt
index 526135f..3d73ed7 100644
--- a/dev/_sources/auto_examples/deep_learning/sg_execution_times.rst.txt
+++ b/dev/_sources/auto_examples/deep_learning/sg_execution_times.rst.txt
@@ -6,10 +6,10 @@
Computation times
=================
-**01:44.325** total execution time for **auto_examples_deep_learning** files:
+**02:33.975** total execution time for **auto_examples_deep_learning** files:
+-------------------------------------------------------------------------------------------------------------+-----------+--------+
-| :ref:`sphx_glr_auto_examples_deep_learning_plot_sgd_solvers.py` (``plot_sgd_solvers.py``) | 01:44.325 | 0.0 MB |
+| :ref:`sphx_glr_auto_examples_deep_learning_plot_sgd_solvers.py` (``plot_sgd_solvers.py``) | 02:33.975 | 0.0 MB |
+-------------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_auto_examples_deep_learning_distributed_flax_imagenet.py` (``distributed_flax_imagenet.py``) | 00:00.000 | 0.0 MB |
+-------------------------------------------------------------------------------------------------------------+-----------+--------+
diff --git a/dev/_sources/auto_examples/fixed_point/plot_anderson_accelerate_gd.rst.txt b/dev/_sources/auto_examples/fixed_point/plot_anderson_accelerate_gd.rst.txt
index 7fddb9a..7311c88 100644
--- a/dev/_sources/auto_examples/fixed_point/plot_anderson_accelerate_gd.rst.txt
+++ b/dev/_sources/auto_examples/fixed_point/plot_anderson_accelerate_gd.rst.txt
@@ -157,7 +157,7 @@ Here `m` denotes the history size, and `K` the frequency of Anderson updates.
.. rst-class:: sphx-glr-timing
- **Total running time of the script:** (0 minutes 2.569 seconds)
+ **Total running time of the script:** (0 minutes 3.292 seconds)
.. _sphx_glr_download_auto_examples_fixed_point_plot_anderson_accelerate_gd.py:
diff --git a/dev/_sources/auto_examples/fixed_point/plot_anderson_wrapper_cd.rst.txt b/dev/_sources/auto_examples/fixed_point/plot_anderson_wrapper_cd.rst.txt
index 5c2cd70..bac7a3a 100644
--- a/dev/_sources/auto_examples/fixed_point/plot_anderson_wrapper_cd.rst.txt
+++ b/dev/_sources/auto_examples/fixed_point/plot_anderson_wrapper_cd.rst.txt
@@ -143,7 +143,7 @@ AISTATS, 2021.
.. rst-class:: sphx-glr-timing
- **Total running time of the script:** (0 minutes 2.866 seconds)
+ **Total running time of the script:** (0 minutes 3.555 seconds)
.. _sphx_glr_download_auto_examples_fixed_point_plot_anderson_wrapper_cd.py:
diff --git a/dev/_sources/auto_examples/fixed_point/plot_picard_ode.rst.txt b/dev/_sources/auto_examples/fixed_point/plot_picard_ode.rst.txt
index 8fcb65f..6037915 100644
--- a/dev/_sources/auto_examples/fixed_point/plot_picard_ode.rst.txt
+++ b/dev/_sources/auto_examples/fixed_point/plot_picard_ode.rst.txt
@@ -152,7 +152,7 @@ integration, but any other integration method can be used.
.. rst-class:: sphx-glr-timing
- **Total running time of the script:** (0 minutes 7.210 seconds)
+ **Total running time of the script:** (0 minutes 9.040 seconds)
.. _sphx_glr_download_auto_examples_fixed_point_plot_picard_ode.py:
diff --git a/dev/_sources/auto_examples/fixed_point/sg_execution_times.rst.txt b/dev/_sources/auto_examples/fixed_point/sg_execution_times.rst.txt
index a60369e..32a027e 100644
--- a/dev/_sources/auto_examples/fixed_point/sg_execution_times.rst.txt
+++ b/dev/_sources/auto_examples/fixed_point/sg_execution_times.rst.txt
@@ -6,14 +6,14 @@
Computation times
=================
-**00:12.645** total execution time for **auto_examples_fixed_point** files:
+**00:15.888** total execution time for **auto_examples_fixed_point** files:
+---------------------------------------------------------------------------------------------------------------+-----------+--------+
-| :ref:`sphx_glr_auto_examples_fixed_point_plot_picard_ode.py` (``plot_picard_ode.py``) | 00:07.210 | 0.0 MB |
+| :ref:`sphx_glr_auto_examples_fixed_point_plot_picard_ode.py` (``plot_picard_ode.py``) | 00:09.040 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------------+-----------+--------+
-| :ref:`sphx_glr_auto_examples_fixed_point_plot_anderson_wrapper_cd.py` (``plot_anderson_wrapper_cd.py``) | 00:02.866 | 0.0 MB |
+| :ref:`sphx_glr_auto_examples_fixed_point_plot_anderson_wrapper_cd.py` (``plot_anderson_wrapper_cd.py``) | 00:03.555 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------------+-----------+--------+
-| :ref:`sphx_glr_auto_examples_fixed_point_plot_anderson_accelerate_gd.py` (``plot_anderson_accelerate_gd.py``) | 00:02.569 | 0.0 MB |
+| :ref:`sphx_glr_auto_examples_fixed_point_plot_anderson_accelerate_gd.py` (``plot_anderson_accelerate_gd.py``) | 00:03.292 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------------+-----------+--------+
| :ref:`sphx_glr_auto_examples_fixed_point_deep_equilibrium_model.py` (``deep_equilibrium_model.py``) | 00:00.000 | 0.0 MB |
+---------------------------------------------------------------------------------------------------------------+-----------+--------+
diff --git a/dev/_sources/changelog.rst.txt b/dev/_sources/changelog.rst.txt
index e58aac0..2e47f05 100644
--- a/dev/_sources/changelog.rst.txt
+++ b/dev/_sources/changelog.rst.txt
@@ -1,9 +1,8 @@
Changelog
=========
-Version 0.9
------------
-
+Version 0.8.1
+-------------
Bug fixes and enhancements
~~~~~~~~~~~~~~~~~~~~~~~~~~
@@ -11,7 +10,16 @@ Bug fixes and enhancements
- Improved example :doc:`/notebooks/deep_learning/resnet_flax` (better accuracy in less time) by Fabian Pedregosa.
- Improved example :doc:`/notebooks/deep_learning/resnet_haiku`, by Fabian Pedregosa.
- Documentation is built as part of the continuous integration workflow, by Fabian Pedregosa
+- Jit update method, by Mathieu Blondel.
+- Linesearch improvements, by Vincent Roulet.
+- GaussNewton and LevenbergMarquardt improvements, by Amir Saadat.
+- Verbose support even when jit=True, by Amir Saadat.
+- Various doc improvements, by Fabian Pedregosa.
+
+Contributors
+~~~~~~~~~~~~
+Amir Saadat, Fabian Pedregosa, Mathieu Blondel, Peter Hawkins, Vincent Roulet.
Version 0.8
-----------
diff --git a/dev/_static/documentation_options.js b/dev/_static/documentation_options.js
index 80dd588..b3ec1c4 100644
--- a/dev/_static/documentation_options.js
+++ b/dev/_static/documentation_options.js
@@ -1,6 +1,6 @@
var DOCUMENTATION_OPTIONS = {
URL_ROOT: document.getElementById("documentation_options").getAttribute('data-url_root'),
- VERSION: '0.8',
+ VERSION: '0.8.1',
LANGUAGE: 'en',
COLLAPSE_INDEX: false,
BUILDER: 'html',
diff --git a/dev/api.html b/dev/api.html
index ad59964..286c96e 100644
--- a/dev/api.html
+++ b/dev/api.html
@@ -4,7 +4,7 @@
- API at a glance — JAXopt 0.8 documentation
+ API at a glance — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/constrained/binary_kernel_svm_with_intercept.html b/dev/auto_examples/constrained/binary_kernel_svm_with_intercept.html
index d983895..4475ee1 100644
--- a/dev/auto_examples/constrained/binary_kernel_svm_with_intercept.html
+++ b/dev/auto_examples/constrained/binary_kernel_svm_with_intercept.html
@@ -4,7 +4,7 @@
- Binary kernel SVM with intercept. — JAXopt 0.8 documentation
+ Binary kernel SVM with intercept. — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/constrained/index.html b/dev/auto_examples/constrained/index.html
index 4e73b55..d01bcbd 100644
--- a/dev/auto_examples/constrained/index.html
+++ b/dev/auto_examples/constrained/index.html
@@ -4,7 +4,7 @@
- Constrained optimization — JAXopt 0.8 documentation
+ Constrained optimization — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/constrained/multiclass_linear_svm.html b/dev/auto_examples/constrained/multiclass_linear_svm.html
index 693c11d..dc71bee 100644
--- a/dev/auto_examples/constrained/multiclass_linear_svm.html
+++ b/dev/auto_examples/constrained/multiclass_linear_svm.html
@@ -4,7 +4,7 @@
- Multiclass linear SVM (without intercept). — JAXopt 0.8 documentation
+ Multiclass linear SVM (without intercept). — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/constrained/nmf.html b/dev/auto_examples/constrained/nmf.html
index c0f9e3f..49a8335 100644
--- a/dev/auto_examples/constrained/nmf.html
+++ b/dev/auto_examples/constrained/nmf.html
@@ -4,7 +4,7 @@
- Non-negative matrix factorizaton (NMF) using alternating minimization. — JAXopt 0.8 documentation
+ Non-negative matrix factorizaton (NMF) using alternating minimization. — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/deep_learning/distributed_flax_imagenet.html b/dev/auto_examples/deep_learning/distributed_flax_imagenet.html
index c54a63c..fc7df48 100644
--- a/dev/auto_examples/deep_learning/distributed_flax_imagenet.html
+++ b/dev/auto_examples/deep_learning/distributed_flax_imagenet.html
@@ -4,7 +4,7 @@
- SPMD ResNet example with Flax and JAXopt. — JAXopt 0.8 documentation
+ SPMD ResNet example with Flax and JAXopt. — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/deep_learning/haiku_vae.html b/dev/auto_examples/deep_learning/haiku_vae.html
index 20a7891..0d41f69 100644
--- a/dev/auto_examples/deep_learning/haiku_vae.html
+++ b/dev/auto_examples/deep_learning/haiku_vae.html
@@ -4,7 +4,7 @@
- VAE example with Haiku and JAXopt. — JAXopt 0.8 documentation
+ VAE example with Haiku and JAXopt. — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/deep_learning/index.html b/dev/auto_examples/deep_learning/index.html
index 386a116..9103a92 100644
--- a/dev/auto_examples/deep_learning/index.html
+++ b/dev/auto_examples/deep_learning/index.html
@@ -4,7 +4,7 @@
- Deep learning — JAXopt 0.8 documentation
+ Deep learning — JAXopt 0.8.1 documentation
diff --git a/dev/auto_examples/deep_learning/plot_sgd_solvers.html b/dev/auto_examples/deep_learning/plot_sgd_solvers.html
index bd39a9a..f7f869d 100644
--- a/dev/auto_examples/deep_learning/plot_sgd_solvers.html
+++ b/dev/auto_examples/deep_learning/plot_sgd_solvers.html
@@ -4,7 +4,7 @@
- Comparison of different SGD algorithms. — JAXopt 0.8 documentation
+ Comparison of different SGD algorithms. — JAXopt 0.8.1 documentation
@@ -209,14 +209,14 @@
Extraction completed...: 0 file [00:00, ? file/s]
-Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
-Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
+Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
+Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
Dl Size...: 0%| | 0/29 [00:00<?, ? MiB/s]
Extraction completed...: 0 file [00:00, ? file/s]
-Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
+Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
Dl Size...: 0%| | 0/29 [00:00<?, ? MiB/s]
@@ -224,364 +224,390 @@
Extraction completed...: 0%| | 0/1 [00:00<?, ? file/s]
-Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 25%|██▌ | 1/4 [00:00<00:00, 3.65 url/s]
+Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 2.44 file/s]
+Dl Completed...: 25%|██▌ | 1/4 [00:00<00:01, 2.46 url/s]
Dl Size...: 0%| | 0/29 [00:00<?, ? MiB/s]
-Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 2.44 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
+Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
Dl Size...: 0%| | 0/29 [00:00<?, ? MiB/s]
-Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+Extraction completed...: 100%|██████████| 1/1 [00:00<00:00, 2.44 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
Dl Size...: 0%| | 0/29 [00:00<?, ? MiB/s]
-Extraction completed...: 50%|█████ | 1/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+Extraction completed...: 50%|█████ | 1/2 [00:00<00:00, 2.44 file/s]
+
+
+Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.85 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.87 url/s]
Dl Size...: 0%| | 0/29 [00:00<?, ? MiB/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
+Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.85 file/s]
+
+Dl Size...: 3%|▎ | 1/29 [00:01<00:37, 1.35s/ MiB]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+
+Dl Size...: 3%|▎ | 1/29 [00:01<00:37, 1.35s/ MiB]
-Dl Size...: 3%|▎ | 1/29 [00:00<00:22, 1.26 MiB/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-Dl Size...: 3%|▎ | 1/29 [00:00<00:22, 1.26 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+Dl Size...: 7%|▋ | 2/29 [00:01<00:36, 1.35s/ MiB]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-Dl Size...: 7%|▋ | 2/29 [00:00<00:21, 1.26 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Size...: 10%|█ | 3/29 [00:01<00:10, 2.46 MiB/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+Dl Size...: 10%|█ | 3/29 [00:01<00:10, 2.46 MiB/s]
-Dl Size...: 10%|█ | 3/29 [00:00<00:20, 1.26 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+Dl Size...: 14%|█▍ | 4/29 [00:01<00:10, 2.46 MiB/s]
-Dl Size...: 14%|█▍ | 4/29 [00:00<00:19, 1.26 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
+Dl Size...: 17%|█▋ | 5/29 [00:01<00:09, 2.46 MiB/s]
-Dl Size...: 17%|█▋ | 5/29 [00:00<00:03, 7.08 MiB/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-Dl Size...: 17%|█▋ | 5/29 [00:00<00:03, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
+Dl Size...: 21%|██ | 6/29 [00:01<00:09, 2.46 MiB/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
-Dl Size...: 21%|██ | 6/29 [00:00<00:03, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Size...: 24%|██▍ | 7/29 [00:01<00:03, 6.34 MiB/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 50%|█████ | 2/4 [00:00<00:00, 3.65 url/s]
+Dl Size...: 24%|██▍ | 7/29 [00:01<00:03, 6.34 MiB/s]
-Dl Size...: 24%|██▍ | 7/29 [00:00<00:03, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
+Dl Size...: 28%|██▊ | 8/29 [00:01<00:03, 6.34 MiB/s]
-Dl Size...: 24%|██▍ | 7/29 [00:00<00:03, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 50%|█████ | 2/4 [00:01<00:00, 3.87 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
+Dl Size...: 31%|███ | 9/29 [00:01<00:03, 6.34 MiB/s]
-Dl Size...: 28%|██▊ | 8/29 [00:00<00:02, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
+Dl Size...: 31%|███ | 9/29 [00:01<00:03, 6.34 MiB/s]
-Dl Size...: 31%|███ | 9/29 [00:00<00:02, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:00<00:00, 3.01 url/s]
+Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 6.34 MiB/s]
-Dl Size...: 34%|███▍ | 10/29 [00:00<00:02, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 2/2 [00:01<00:00, 3.85 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 2/2 [00:00<00:00, 3.61 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 6.34 MiB/s]
-Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 7.08 MiB/s]
+Extraction completed...: 67%|██████▋ | 2/3 [00:01<00:00, 3.85 file/s]
-Extraction completed...: 67%|██████▋ | 2/3 [00:01<00:00, 3.61 file/s]
+Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 6.34 MiB/s]
-Dl Size...: 34%|███▍ | 10/29 [00:01<00:02, 7.08 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
+Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 9.96 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 14.68 MiB/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 9.96 MiB/s]
-Dl Size...: 38%|███▊ | 11/29 [00:01<00:01, 14.68 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 41%|████▏ | 12/29 [00:01<00:01, 9.96 MiB/s]
-Dl Size...: 41%|████▏ | 12/29 [00:01<00:01, 14.68 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 45%|████▍ | 13/29 [00:01<00:01, 10.66 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Dl Size...: 45%|████▍ | 13/29 [00:01<00:01, 14.68 MiB/s]
+Dl Size...: 45%|████▍ | 13/29 [00:01<00:01, 10.66 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Dl Size...: 48%|████▊ | 14/29 [00:01<00:01, 14.68 MiB/s]
+Dl Size...: 48%|████▊ | 14/29 [00:01<00:01, 10.66 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
+Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 1.53 url/s]
-Dl Size...: 52%|█████▏ | 15/29 [00:01<00:00, 19.13 MiB/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 52%|█████▏ | 15/29 [00:01<00:01, 10.66 MiB/s]
-Dl Size...: 52%|█████▏ | 15/29 [00:01<00:00, 19.13 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 1.46 file/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 55%|█████▌ | 16/29 [00:02<00:01, 12.58 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Dl Size...: 55%|█████▌ | 16/29 [00:01<00:00, 19.13 MiB/s]
+Dl Size...: 55%|█████▌ | 16/29 [00:02<00:01, 12.58 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Dl Size...: 59%|█████▊ | 17/29 [00:01<00:00, 19.13 MiB/s]
+Dl Size...: 59%|█████▊ | 17/29 [00:02<00:00, 12.58 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
-Dl Size...: 62%|██████▏ | 18/29 [00:01<00:00, 19.13 MiB/s]
+Dl Size...: 62%|██████▏ | 18/29 [00:02<00:00, 12.72 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+Dl Size...: 62%|██████▏ | 18/29 [00:02<00:00, 12.72 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Size...: 66%|██████▌ | 19/29 [00:01<00:00, 20.41 MiB/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Dl Size...: 66%|██████▌ | 19/29 [00:01<00:00, 20.41 MiB/s]
+Dl Size...: 66%|██████▌ | 19/29 [00:02<00:00, 12.72 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Dl Size...: 69%|██████▉ | 20/29 [00:01<00:00, 20.41 MiB/s]
+Dl Size...: 69%|██████▉ | 20/29 [00:02<00:00, 12.72 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
-Dl Size...: 72%|███████▏ | 21/29 [00:01<00:00, 20.41 MiB/s]
+Dl Size...: 72%|███████▏ | 21/29 [00:02<00:00, 14.21 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+Dl Size...: 72%|███████▏ | 21/29 [00:02<00:00, 14.21 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-Dl Size...: 76%|███████▌ | 22/29 [00:01<00:00, 20.41 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+Dl Size...: 76%|███████▌ | 22/29 [00:02<00:00, 14.21 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Size...: 79%|███████▉ | 23/29 [00:01<00:00, 21.60 MiB/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
-Dl Size...: 79%|███████▉ | 23/29 [00:01<00:00, 21.60 MiB/s]
+Dl Size...: 79%|███████▉ | 23/29 [00:02<00:00, 14.07 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+Dl Size...: 79%|███████▉ | 23/29 [00:02<00:00, 14.07 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-Dl Size...: 83%|████████▎ | 24/29 [00:01<00:00, 21.60 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+Dl Size...: 83%|████████▎ | 24/29 [00:02<00:00, 14.07 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-Dl Size...: 86%|████████▌ | 25/29 [00:01<00:00, 21.60 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Size...: 86%|████████▌ | 25/29 [00:02<00:00, 13.71 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 86%|████████▌ | 25/29 [00:02<00:00, 13.71 MiB/s]
-Dl Size...: 90%|████████▉ | 26/29 [00:01<00:00, 21.60 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
+Dl Size...: 90%|████████▉ | 26/29 [00:02<00:00, 13.71 MiB/s]
-Dl Size...: 93%|█████████▎| 27/29 [00:01<00:00, 25.00 MiB/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-Dl Size...: 93%|█████████▎| 27/29 [00:01<00:00, 25.00 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
+Dl Size...: 93%|█████████▎| 27/29 [00:02<00:00, 13.71 MiB/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
-Dl Size...: 97%|█████████▋| 28/29 [00:01<00:00, 25.00 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Size...: 97%|█████████▋| 28/29 [00:02<00:00, 15.21 MiB/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 75%|███████▌ | 3/4 [00:01<00:00, 3.01 url/s]
+Dl Size...: 97%|█████████▋| 28/29 [00:02<00:00, 15.21 MiB/s]
-Dl Size...: 100%|██████████| 29/29 [00:01<00:00, 25.00 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 75%|███████▌ | 3/4 [00:02<00:00, 1.53 url/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 100%|██████████| 4/4 [00:01<00:00, 2.13 url/s]
-Dl Completed...: 100%|██████████| 4/4 [00:01<00:00, 2.13 url/s]
+Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 15.21 MiB/s]
-Dl Size...: 100%|██████████| 29/29 [00:01<00:00, 25.00 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 1.13 url/s]
+Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 1.13 url/s]
-Extraction completed...: 100%|██████████| 3/3 [00:01<00:00, 2.79 file/s]
-Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 2.13 url/s]
+Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 15.21 MiB/s]
-Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 25.00 MiB/s]
+Extraction completed...: 100%|██████████| 3/3 [00:02<00:00, 1.46 file/s]
+Dl Completed...: 100%|██████████| 4/4 [00:03<00:00, 1.13 url/s]
-Extraction completed...: 75%|███████▌ | 3/4 [00:02<00:00, 2.79 file/s]
+Dl Size...: 100%|██████████| 29/29 [00:03<00:00, 15.21 MiB/s]
-Extraction completed...: 100%|██████████| 4/4 [00:02<00:00, 1.70 file/s]
-Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 2.13 url/s]
+Extraction completed...: 75%|███████▌ | 3/4 [00:03<00:00, 1.46 file/s]
-Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 25.00 MiB/s]
+Extraction completed...: 100%|██████████| 4/4 [00:03<00:00, 1.06s/ file]
+Dl Completed...: 100%|██████████| 4/4 [00:03<00:00, 1.13 url/s]
-Extraction completed...: 100%|██████████| 4/4 [00:02<00:00, 1.70 file/s]
-Extraction completed...: 100%|██████████| 4/4 [00:02<00:00, 1.93 file/s]
+Dl Size...: 100%|██████████| 29/29 [00:03<00:00, 15.21 MiB/s]
-Dl Size...: 100%|██████████| 29/29 [00:02<00:00, 14.00 MiB/s]
-Dl Completed...: 100%|██████████| 4/4 [00:02<00:00, 1.93 url/s]
+Extraction completed...: 100%|██████████| 4/4 [00:03<00:00, 1.06s/ file]
+Extraction completed...: 100%|██████████| 4/4 [00:03<00:00, 1.18 file/s]
+
+Dl Size...: 100%|██████████| 29/29 [00:03<00:00, 8.56 MiB/s]
+
+Dl Completed...: 100%|██████████| 4/4 [00:03<00:00, 1.18 url/s]
Generating splits...: 0%| | 0/2 [00:00<?, ? splits/s]
Generating train examples...: 0%| | 0/60000 [00:00<?, ? examples/s]
-Generating train examples...: 3%|▎ | 1690/60000 [00:01<00:34, 1689.98 examples/s]
+Generating train examples...: 2%|▏ | 1285/60000 [00:01<00:45, 1284.71 examples/s]
+
+Generating train examples...: 5%|▍ | 2881/60000 [00:02<00:38, 1467.71 examples/s]
-Generating train examples...: 6%|▌ | 3567/60000 [00:02<00:31, 1799.89 examples/s]
+Generating train examples...: 7%|▋ | 4488/60000 [00:03<00:36, 1531.03 examples/s]
-Generating train examples...: 9%|▉ | 5437/60000 [00:03<00:29, 1831.69 examples/s]
+Generating train examples...: 10%|█ | 6094/60000 [00:04<00:34, 1560.59 examples/s]
-Generating train examples...: 12%|█▏ | 7308/60000 [00:04<00:28, 1847.19 examples/s]
+Generating train examples...: 13%|█▎ | 7706/60000 [00:05<00:33, 1579.07 examples/s]
-Generating train examples...: 15%|█▌ | 9180/60000 [00:05<00:27, 1856.03 examples/s]
+Generating train examples...: 16%|█▌ | 9312/60000 [00:06<00:31, 1588.05 examples/s]
-Generating train examples...: 18%|█▊ | 11053/60000 [00:06<00:26, 1861.67 examples/s]
+Generating train examples...: 18%|█▊ | 10904/60000 [00:07<00:30, 1589.29 examples/s]
-Generating train examples...: 22%|██▏ | 12924/60000 [00:07<00:25, 1864.54 examples/s]
+Generating train examples...: 21%|██ | 12511/60000 [00:08<00:29, 1594.64 examples/s]
-Generating train examples...: 25%|██▍ | 14798/60000 [00:08<00:24, 1867.37 examples/s]
+Generating train examples...: 24%|██▎ | 14120/60000 [00:09<00:28, 1598.92 examples/s]
-Generating train examples...: 28%|██▊ | 16671/60000 [00:09<00:23, 1869.05 examples/s]
+Generating train examples...: 26%|██▌ | 15731/60000 [00:10<00:27, 1602.37 examples/s]
-Generating train examples...: 31%|███ | 18542/60000 [00:10<00:22, 1869.37 examples/s]
+Generating train examples...: 29%|██▉ | 17341/60000 [00:11<00:26, 1604.65 examples/s]
-Generating train examples...: 34%|███▍ | 20415/60000 [00:11<00:21, 1870.20 examples/s]
+Generating train examples...: 32%|███▏ | 18946/60000 [00:12<00:25, 1602.91 examples/s]
-Generating train examples...: 37%|███▋ | 22290/60000 [00:12<00:20, 1871.52 examples/s]
+Generating train examples...: 34%|███▍ | 20552/60000 [00:13<00:24, 1603.74 examples/s]
-Generating train examples...: 40%|████ | 24162/60000 [00:13<00:19, 1865.31 examples/s]
+Generating train examples...: 37%|███▋ | 22160/60000 [00:14<00:23, 1605.01 examples/s]
-Generating train examples...: 43%|████▎ | 26033/60000 [00:14<00:18, 1866.98 examples/s]
+Generating train examples...: 40%|███▉ | 23769/60000 [00:15<00:22, 1605.96 examples/s]
-Generating train examples...: 47%|████▋ | 27910/60000 [00:15<00:17, 1869.86 examples/s]
+Generating train examples...: 42%|████▏ | 25375/60000 [00:16<00:21, 1602.23 examples/s]
-Generating train examples...: 50%|████▉ | 29788/60000 [00:16<00:16, 1872.02 examples/s]
+Generating train examples...: 45%|████▍ | 26983/60000 [00:17<00:20, 1603.86 examples/s]
-Generating train examples...: 53%|█████▎ | 31668/60000 [00:17<00:15, 1874.25 examples/s]
+Generating train examples...: 48%|████▊ | 28590/60000 [00:18<00:19, 1604.50 examples/s]
-Generating train examples...: 56%|█████▌ | 33544/60000 [00:18<00:14, 1874.63 examples/s]
+Generating train examples...: 50%|█████ | 30196/60000 [00:19<00:18, 1604.81 examples/s]
-Generating train examples...: 59%|█████▉ | 35421/60000 [00:19<00:13, 1875.26 examples/s]
+Generating train examples...: 53%|█████▎ | 31801/60000 [00:20<00:17, 1604.09 examples/s]
-Generating train examples...: 62%|██████▏ | 37297/60000 [00:20<00:12, 1875.43 examples/s]
+Generating train examples...: 56%|█████▌ | 33407/60000 [00:21<00:16, 1604.40 examples/s]
-Generating train examples...: 65%|██████▌ | 39175/60000 [00:21<00:11, 1876.16 examples/s]
+Generating train examples...: 58%|█████▊ | 35014/60000 [00:22<00:15, 1605.06 examples/s]
-Generating train examples...: 68%|██████▊ | 41055/60000 [00:22<00:10, 1877.14 examples/s]
+Generating train examples...: 61%|██████ | 36620/60000 [00:23<00:14, 1604.24 examples/s]
-Generating train examples...: 72%|███████▏ | 42935/60000 [00:23<00:09, 1877.82 examples/s]
+Generating train examples...: 64%|██████▎ | 38225/60000 [00:24<00:13, 1604.44 examples/s]
-Generating train examples...: 75%|███████▍ | 44813/60000 [00:24<00:08, 1877.17 examples/s]
+Generating train examples...: 66%|██████▋ | 39830/60000 [00:25<00:12, 1604.24 examples/s]
-Generating train examples...: 78%|███████▊ | 46691/60000 [00:25<00:07, 1876.61 examples/s]
+Generating train examples...: 69%|██████▉ | 41435/60000 [00:26<00:11, 1603.83 examples/s]
-Generating train examples...: 81%|████████ | 48568/60000 [00:26<00:06, 1875.08 examples/s]
+Generating train examples...: 72%|███████▏ | 43043/60000 [00:27<00:10, 1604.79 examples/s]
-Generating train examples...: 84%|████████▍ | 50449/60000 [00:27<00:05, 1876.75 examples/s]
+Generating train examples...: 74%|███████▍ | 44648/60000 [00:28<00:09, 1603.42 examples/s]
-Generating train examples...: 87%|████████▋ | 52326/60000 [00:28<00:04, 1875.66 examples/s]
+Generating train examples...: 77%|███████▋ | 46252/60000 [00:29<00:08, 1603.22 examples/s]
-Generating train examples...: 90%|█████████ | 54205/60000 [00:29<00:03, 1876.52 examples/s]
+Generating train examples...: 80%|███████▉ | 47857/60000 [00:30<00:07, 1603.54 examples/s]
-Generating train examples...: 93%|█████████▎| 56082/60000 [00:30<00:02, 1876.61 examples/s]
+Generating train examples...: 82%|████████▏ | 49461/60000 [00:31<00:06, 1599.84 examples/s]
-Generating train examples...: 97%|█████████▋| 57960/60000 [00:31<00:01, 1876.76 examples/s]
+Generating train examples...: 85%|████████▌ | 51061/60000 [00:32<00:05, 1599.78 examples/s]
-Generating train examples...: 100%|█████████▉| 59837/60000 [00:32<00:00, 1876.49 examples/s]
+Generating train examples...: 88%|████████▊ | 52661/60000 [00:33<00:04, 1597.94 examples/s]
+Generating train examples...: 90%|█████████ | 54263/60000 [00:34<00:03, 1599.08 examples/s]
+Generating train examples...: 93%|█████████▎| 55863/60000 [00:35<00:02, 1599.19 examples/s]
-Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteD287AU/fashion_mnist-train.tfrecord*...: 0%| | 0/60000 [00:00<?, ? examples/s]
+Generating train examples...: 96%|█████████▌| 57464/60000 [00:36<00:01, 1599.43 examples/s]
-Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteD287AU/fashion_mnist-train.tfrecord*...: 46%|████▋ | 27864/60000 [00:00<00:00, 278617.68 examples/s]
+Generating train examples...: 98%|█████████▊| 59064/60000 [00:37<00:00, 1598.35 examples/s]
-Generating splits...: 50%|█████ | 1/2 [00:32<00:32, 32.31s/ splits]
+
+Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-train.tfrecord*...: 0%| | 0/60000 [00:00<?, ? examples/s]
+
+Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-train.tfrecord*...: 30%|███ | 18179/60000 [00:00<00:00, 181777.17 examples/s]
+
+Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-train.tfrecord*...: 93%|█████████▎| 55686/60000 [00:00<00:00, 295461.61 examples/s]
+
+
+Generating splits...: 50%|█████ | 1/2 [00:37<00:37, 37.87s/ splits]
Generating test examples...: 0%| | 0/10000 [00:00<?, ? examples/s]
-Generating test examples...: 19%|█▊ | 1860/10000 [00:01<00:04, 1859.89 examples/s]
+Generating test examples...: 16%|█▌ | 1580/10000 [00:01<00:05, 1579.80 examples/s]
+
+Generating test examples...: 32%|███▏ | 3194/10000 [00:02<00:04, 1599.55 examples/s]
-Generating test examples...: 37%|███▋ | 3747/10000 [00:02<00:03, 1875.63 examples/s]
+Generating test examples...: 48%|████▊ | 4804/10000 [00:03<00:03, 1604.15 examples/s]
-Generating test examples...: 56%|█████▋ | 5635/10000 [00:03<00:02, 1880.82 examples/s]
+Generating test examples...: 64%|██████▍ | 6418/10000 [00:04<00:02, 1607.73 examples/s]
-Generating test examples...: 75%|███████▌ | 7523/10000 [00:04<00:01, 1883.31 examples/s]
+Generating test examples...: 80%|████████ | 8031/10000 [00:05<00:01, 1609.37 examples/s]
-Generating test examples...: 94%|█████████▍| 9411/10000 [00:05<00:00, 1884.99 examples/s]
+Generating test examples...: 96%|█████████▋| 9645/10000 [00:06<00:00, 1610.82 examples/s]
-Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteD287AU/fashion_mnist-test.tfrecord*...: 0%| | 0/10000 [00:00<?, ? examples/s]
+Shuffling /home/runner/tensorflow_datasets/fashion_mnist/3.0.1.incompleteVJPFTA/fashion_mnist-test.tfrecord*...: 0%| | 0/10000 [00:00<?, ? examples/s]
-Generating splits...: 100%|██████████| 2/2 [00:37<00:00, 16.45s/ splits]
+Generating splits...: 100%|██████████| 2/2 [00:44<00:00, 19.28s/ splits]
Dataset fashion_mnist downloaded and prepared to /home/runner/tensorflow_datasets/fashion_mnist/3.0.1. Subsequent calls will reuse this data.
[Polyak SGD] ( 0), error=0.272, loss=2.301, stepsize=1.000000
@@ -590,7 +616,7 @@
[Polyak SGD] ( 60), error=1.689, loss=0.831, stepsize=0.291400
[Polyak SGD] ( 80), error=1.325, loss=1.168, stepsize=0.665042
[Polyak SGD] ( 100), error=1.382, loss=0.916, stepsize=0.479756
-101 iterations of Polyak SGD took 15.03s
+101 iterations of Polyak SGD took 19.83s
[Armijo SGD] ( 0), error=0.227, loss=2.313, stepsize=1.000000
[Armijo SGD] ( 20), error=2.221, loss=1.331, stepsize=0.065971
@@ -598,7 +624,7 @@
[Armijo SGD] ( 60), error=3.499, loss=1.059, stepsize=0.027022
[Armijo SGD] ( 80), error=1.456, loss=0.863, stepsize=0.021617
[Armijo SGD] ( 100), error=1.689, loss=0.819, stepsize=0.021617
-101 iterations of Armijo SGD took 17.76s
+101 iterations of Armijo SGD took 23.17s
[RMSProp] ( 0), error=0.223, loss=2.299
[RMSProp] ( 20), error=0.869, loss=1.913
@@ -606,7 +632,7 @@
[RMSProp] ( 60), error=1.115, loss=1.211
[RMSProp] ( 80), error=1.627, loss=1.088
[RMSProp] ( 100), error=1.581, loss=1.002
-101 iterations of RMSProp took 13.94s
+101 iterations of RMSProp took 17.58s
[SGD] ( 0), error=0.190, loss=2.306
[SGD] ( 20), error=0.384, loss=2.236
@@ -614,7 +640,7 @@
[SGD] ( 60), error=1.222, loss=1.092
[SGD] ( 80), error=1.637, loss=1.146
[SGD] ( 100), error=4.393, loss=1.671
-101 iterations of SGD took 13.78s
+101 iterations of SGD took 17.34s
@@ -799,7 +825,7 @@
main ( sys . argv )
-Total running time of the script: (1 minutes 44.325 seconds)
+Total running time of the script: (2 minutes 33.975 seconds)