From 7e8c661d1d91d415c0222a96f791ee64b7ce7a23 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jaan=20L=C4=B1=20=E6=9D=8E=20PhD?= Date: Wed, 24 Apr 2024 07:45:21 -0600 Subject: [PATCH] Update README.md --- README.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/README.md b/README.md index cf47eef..012cef1 100644 --- a/README.md +++ b/README.md @@ -69,3 +69,8 @@ Total time: 2.350 minutes 1. Run `python train_variational_autoencoder_tensorflow.py` 2. Install imagemagick (homebrew for Mac: https://formulae.brew.sh/formula/imagemagick or Chocolatey in Windows: https://community.chocolatey.org/packages/imagemagick.app) 3. Go to the directory where the jpg files are saved, and run the imagemagick command to generate the .gif: `convert -delay 20 -loop 0 *.jpg latent-space.gif` +4. + +## TODO (help needed - feel free to send a PR!) +- add multiple GPU / TPU option +- add jaxtyping support for PyTorch and Jax implementations :) for runtime static type checking (using @beartype decorators)