Markus Enzweiler
Markus Enzweiler
- Tested with mlx 0.0.9 - Requirements and documentation updated - Style checked - Rebased and forced-pushed I think it is ready for review.
Tested with mlx-0.0.11, rebased and force-pushed
@awni Yes. I have made the minor changes requested above and it's ready for review.
Thanks @awni. I have also started a training run.
@awni Great! Thanks for merging. I am getting a throughput of approx. 485 im/s on my M1 which is quite a bit faster than 410 im/s with previous mlx versions....
I guess you cannot use both at the moment with MLX. You would need to re-implement / port the models you need over to MLX.
For MaxPooling, also have a look at https://github.com/robertmccraith/mimm/blob/main/mimm/layers/max_pool.py
I have to pass on that, sorry. But I will follow your progress and am happy to try out and test your implementation.