From 37ab915736ba44cb12a2d0d1c0a9ef112cecdcc8 Mon Sep 17 00:00:00 2001 From: Jascha Sohl-Dickstein Date: Thu, 18 Jan 2024 21:40:17 -0800 Subject: [PATCH] Update README.md --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 8eb9cd0..69cd9d6 100644 --- a/README.md +++ b/README.md @@ -6,7 +6,7 @@ *learned\_optimization* is a research codebase for training, designing, evaluating, and applying learned optimizers, and for meta-training of dynamical systems more broadly. It implements hand-designed and learned optimizers, tasks to meta-train and meta-test them, and outer-training algorithms such as ES, PES, and truncated backprop through time. -To get started see our [documentation](https://learned-optimization.readthedocs.io/en/latest/). +There is a third-party (alpha) [PyTorch wrapper for VeLO](https://github.com/janEbert/PyTorch-VeLO) (the current SOTA learned optimizer in this repository) by Jan Ebert. ## Quick Start Colab Notebooks Our [documentation](https://learned-optimization.readthedocs.io/en/latest/) can also be run as colab notebooks! We recommend running these notebooks with a free accelerator (TPU or GPU) in colab (go to `Runtime` -> `Change runtime type`).