mlx
mlx copied to clipboard
[Feature Request] Array pickle support
Add pickle support to array through pybind, docs can be viewed here. This is a pre-req for #300
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 of managing the state of the array (a lot of it is not currently publicly accessible). If we want to keep it that way, we probably need to do a contiguous copy first.
Also it probably should not be ok to pickle an array used in tracing the graph in a function transform.
Is anyone actively working on this?
If no one is working on it, I can give it a shot
Sounds great!! I will assign to you!
Anyone working on this?
@chunyang-wen any progress?
@dc-dc-dc Sorry. I am on holidays. Will work on it.
I will remove myself as people are working on it. Sorry for the delay.