[FEATURE] adding MPS GRUCell for efficiency

ml-explore / mlx

MLX: An array framework for Apple silicon

https://ml-explore.github.io/mlx/

MIT License

16.82k stars 968 forks source link

[FEATURE] adding MPS GRUCell for efficiency #1500

Open thegodone opened 2 days ago

thegodone commented 2 days ago

Describe the bug

This is related to the #1499 , the GRUCell is not implemented and the GRU version is also not optimized on MPS compare to pytorch and tensorflow variant.

angeloskath commented 1 day ago

I don't think this is related to #1499 but we definitely need to speed up our RNN implementations.