mlx-examples/cvae/requirements.txt
Awni Hannun f45a1ab83c
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

5 lines
31 B
Plaintext

mlx>=0.2
mlx-data
numpy
Pillow