zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 04:14:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update a few examples to use compile (#420)

Browse Source 
* 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
This commit is contained in:
Awni Hannun
2024-02-08 13:00:41 -08:00
committed by GitHub
parent da7adae5ec
commit f45a1ab83c
17 changed files with 164 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cvae/requirements.txt
View File

@@ -1,4 +1,4 @@
mlx>=0.0.9
mlx>=0.2
mlx-data
numpy
Pillow