Progress Log - Pradyot

[Pradyot_Ranjan]

Hi,

As part of FOSS United’s grants program, I’ll be sharing biweekly updates here! Some of the topics I’ll be working on include:

• Work on extending MLX work to array-api-extra, Scipy, etc.
• Fix the missing/failing MLX array api tests. and increase adherence to the standards.
• Work on supporting `torch.compile` in array-api-extra
• `torch.mps` array api support
• Array api work in finch-tensor library (If we could manage the above work)

Thank you so much @ansh @rahulporuri

[Soham_Kulkarni]

cool to see someone else working on a similar SoC problem statement! i have a m4 available for testing, lmk how i can help

[Pradyot_Ranjan]

Hi,

Here are my updates so far.

• The biggest update is adding MLX support to array-api-extra. This would enable libraries using MLX as array providers to benefit from the extra functions (functions outside the Array API Standards) in array-api-extra. This is being done in ENH: Adding MLX by prady0t · Pull Request #1008 · data-apis/array-api-extra · GitHub . Currently, this PR requires some maintainers’ feedback, and we also need to check what things we need to port to the array-api-compat (the compatibility layer library) before proceeding further.
• Some efforts have also been made towards TST: Use assert functions from `array-api-extra` by prady0t · Pull Request #34868 · scikit-learn/scikit-learn · GitHub; it still requires maintainers’ feedback. But overall, it’s very close.
• MLX now runs array-api-tests in the CI! This is a great milestone as it helps keep track of how much closer we are towards 100% compatibility. We maintain a list of skipped tests, and every time someone adds some functionality, it’s removed from the list. See: Re-introduce Array API Tests suite by prady0t · Pull Request #4433 · ml-explore/mlx · GitHub. One example is my recent fix for the broadcast_arrays function here.

Note: I had limited availability the last few weeks, and this will continue for another 2 weeks (I’m in between onboarding, travel, festivals, etc.). There will be much more progress to share after that!

Thanks for reading.