Update optimize_2d_momentum_bumpy_torch.py

v0lta · web-flow · commit 2c07a65fe5ed · 2025-04-28T11:52:16.000+02:00
diff --git a/src/optimize_2d_momentum_bumpy_torch.py b/src/optimize_2d_momentum_bumpy_torch.py
@@ -1,4 +1,4 @@
-"""Implement 2d gradient descent using jax."""
+"""Implement 2d gradient descent."""
 
 import numpy as np
 import torch as th

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-"""Implement 2d gradient descent using jax."""`
	`1`	`+"""Implement 2d gradient descent."""`
`2`	`2`
`3`	`3`	`import numpy as np`
`4`	`4`	`import torch as th`