| ## Proposed changes | |
| Please include a description of the problem or feature this PR is addressing. If there is a corresponding issue, include the issue #. | |
| ## Checklist | |
| Put an `x` in the boxes that apply. | |
| - [ ] I have read the [CONTRIBUTING](https://github.com/ml-explore/mlx/blob/main/CONTRIBUTING.md) 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) | |