aboutsummaryrefslogtreecommitdiff
path: root/misc/py-schedulefree
diff options
context:
space:
mode:
authorYuri Victorovich <yuri@FreeBSD.org>2025-03-29 06:48:56 +0000
committerYuri Victorovich <yuri@FreeBSD.org>2025-03-29 16:27:46 +0000
commitd9e91d68282c6f50cd9c4eaeb3250bba93829b0d (patch)
tree3062907dd4e1eb2f28273cec6d91afdfd020f7e3 /misc/py-schedulefree
parent8866cf92aeec45f659e030fde5842a241d1ad5d4 (diff)
Diffstat (limited to 'misc/py-schedulefree')
-rw-r--r--misc/py-schedulefree/Makefile25
-rw-r--r--misc/py-schedulefree/distinfo3
-rw-r--r--misc/py-schedulefree/pkg-descr11
3 files changed, 39 insertions, 0 deletions
diff --git a/misc/py-schedulefree/Makefile b/misc/py-schedulefree/Makefile
new file mode 100644
index 000000000000..e02ac03e7402
--- /dev/null
+++ b/misc/py-schedulefree/Makefile
@@ -0,0 +1,25 @@
+PORTNAME= schedulefree
+DISTVERSION= 1.4.1
+CATEGORIES= misc python # machine-learning
+MASTER_SITES= PYPI
+PKGNAMEPREFIX= ${PYTHON_PKGNAMEPREFIX}
+
+MAINTAINER= yuri@FreeBSD.org
+COMMENT= Schedule free learning in PyTorch
+WWW= https://github.com/facebookresearch/schedule_free
+
+LICENSE= APACHE20
+LICENSE_FILE= ${WRKSRC}/LICENSE
+
+BUILD_DEPENDS= ${PYTHON_PKGNAMEPREFIX}hatchling>0:devel/py-hatchling@${PY_FLAVOR}
+RUN_DEPENDS= ${PYTHON_PKGNAMEPREFIX}pytorch>0:misc/py-pytorch@${PY_FLAVOR} \
+ ${PYTHON_PKGNAMEPREFIX}typing-extensions>0:devel/py-typing-extensions@${PY_FLAVOR}
+
+USES= python
+USE_PYTHON= pep517 autoplist pytest
+
+NO_ARCH= yes
+
+# most tests fail: AssertionError: Torch not compiled with CUDA enabled, see https://github.com/facebookresearch/schedule_free/issues/65
+
+.include <bsd.port.mk>
diff --git a/misc/py-schedulefree/distinfo b/misc/py-schedulefree/distinfo
new file mode 100644
index 000000000000..ae3b5ab309fd
--- /dev/null
+++ b/misc/py-schedulefree/distinfo
@@ -0,0 +1,3 @@
+TIMESTAMP = 1742853003
+SHA256 (schedulefree-1.4.1.tar.gz) = 69ef25601d1fc0d8dd00cb36f9af78833f88b7846f1bb6ddecc9f144f3e9f7cb
+SIZE (schedulefree-1.4.1.tar.gz) = 29281
diff --git a/misc/py-schedulefree/pkg-descr b/misc/py-schedulefree/pkg-descr
new file mode 100644
index 000000000000..723ec22bfa88
--- /dev/null
+++ b/misc/py-schedulefree/pkg-descr
@@ -0,0 +1,11 @@
+Schedulefree is a Schedule-Free optimizer in PyTorch.
+
+We provide several Schedule-Free optimizer implementations:
+
+* SGDScheduleFree and SGDScheduleFreeReference: Schedule-free variants of SGD
+* AdamWScheduleFree and AdamWScheduleFreeReference: Schedule-free variants
+ of AdamW
+* RAdamScheduleFree: Schedule-free variant of RAdam, which eliminates the need
+ for both learning rate scheduling and warmup (implementation community
+ contributed)
+* Experimental ScheduleFreeWrapper to combine with other optimizers