Awni Hannun
Awni Hannun
Spoke too soon. This is not actually fixed (se #366 )
Thanks for adding this. I was thinking we should use concatenate and have this be part of the compute graph rather than force an eval. Concretely when you do `mx.array([a1,...
Hi @Alex-X-W , There's two things I think we need to fix with this: 1. The main point to using stack is to avoid needing to eval the input arrays....
PS checked cpu/gpu tests pass + formatting passes
We can close this as the latest MLX works with `einops.array_api` back-end.
Yea on our end it's not a top priority. But if someone is interested in contributing this, that would be great. To start it could be a separate community maintained...
Thanks for making the issue! I don't see any reason we shouldn't let an array be pickle-able, so I support adding this. There may be some thorny issues in terms...
Is anyone actively working on this?
Sounds great!! I will assign to you!
To be honest this wasn't a very intentional design decision. Pybind11 by default translates std::vector to Python list so that's why it is the way it is. I believe it's...