Add printoptions

[ChristophePRAT]

Proposed changes

Add minimal print options similar to numpy's set_printoptions.

This PR should fix #471.

Quick example

import mlx.core as mx

a = mx.array([0.3668831623])

print(a) # 0.3668831623

with mx.printoptions(precision=2):
    print(a) # 0.37

print(a) # 0.3668831623

mx.set_printoptions(precision=4)
print(a) # 0.3669
print(a) # 0.3669

Checklist

Put an x in the boxes that apply.

• I have read the CONTRIBUTING document
• I have run pre-commit run --all-files to format my code / installed pre-commit prior to committing changes
• I have added tests that prove my fix is effective or that my feature works
• I have updated the necessary documentation (if needed)

[ChristophePRAT]

Thank you very much for your feedback. I have incorporated all of the changes that you mentioned in the previous commits. Let me know if I missed something.

[zcbenz]

Can you rebase this PR on latest main branch and exclude unrelated commits?

[ChristophePRAT]

Sorry for the git mistakes. This should be solved.

[zcbenz]

Looks good to me, thanks!

[ChristophePRAT]

Thank you very much for the review and the feedback, really appreciate your time!