mlx-examples/cvae/requirements.txt

mlx>=0.2
mlx-data
numpy
Pillow
Update a few examples to use compile (#420) * update a few examples to use compile * update mnist * add compile to vae and rename some stuff for simplicity * update reqs * use state in eval * GCN example with RNG + dropout * add a bit of prefetching 2024-02-08 13:00:41 -08:00			`mlx>=0.2`
Example of a Convolutional Variational Autoencoder (CVAE) on MNIST (#264) * initial commit * style fixes * update of ACKNOWLEDGMENTS * fixed comment * minor refactoring; removed unused imports * added cifar and cvae to top-level README.md * removed mention of cuda/mps in argparse * fixed training status output * load_weights() with strict=True * pretrained model update * fixed imports and style * requires mlx>=0.0.9 * updated with results using mlx 0.0.9 * removed mention of private repo * simplify and combine to one file, more consistency with other exmaples * few more nits * nits * spell * format --------- Co-authored-by: Awni Hannun <awni@apple.com> 2024-02-07 05:02:27 +01:00			`mlx-data`
			`numpy`
			`Pillow`