[Feature] Support pickling array for bfloat16

[awni]

Leaving this here in case someone wants to pick it up.

[MacOS]

I would be interested.

[awni]

Great!! Send a PR?

[MacOS]

I will! :)

May I ask you to assign the issue to me so that it appears on my GitHub landing page and reminds me to do it ASAP?

[AvikantSrivastava]

@MacOS are you still working on this? If not I would be happy to take it up :)
cc: @awni

[awni]

Go for it @AvikantSrivastava

[AvikantSrivastava]

@awni I was referring to this PR https://github.com/ml-explore/mlx/pull/713/files for the implementation of pickle bindings. It seems that this piece of code was later removed, can you point me to the pickling implementation?

[awni]

Just that: https://github.com/ml-explore/mlx/blob/main/python/src/array.cpp#L672-L684