mlx icon indicating copy to clipboard operation
mlx copied to clipboard
verified publisher icon ml-explore

[FEATURE] adding MPS GRUCell for efficiency

Open thegodone opened this issue 1 year ago • 1 comments

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.

thegodone avatar Oct 18 '24 11:10 thegodone

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

angeloskath avatar Oct 18 '24 23:10 angeloskath