mlx-examples/llms/mlx_lm/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
45 B
Plaintext

mlx>=0.1
numpy
transformers>=4.37.0
protobuf