Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature suggestion] apple mlx support #295

Closed
lukaemon opened this issue Dec 6, 2023 · 3 comments
Closed

[Feature suggestion] apple mlx support #295

lukaemon opened this issue Dec 6, 2023 · 3 comments

Comments

@lukaemon
Copy link

lukaemon commented Dec 6, 2023

Hi,
MLX: An array framework for Apple silicon happened today.
I hope it's the precursor to open up Neural Engine.

Would love to see einops supports mlx.

@altaic
Copy link

altaic commented Aug 6, 2024

FYI, einsum has landed in the mlx main branch, and I'd expect it to be released very soon (likely this Thursday). Looking forward to using einops w/ mlx!

@altaic
Copy link

altaic commented Aug 6, 2024

Also see ml-explore/mlx#1289 for details on the array api support.

@arogozhnikov
Copy link
Owner

that's great news @altaic ! Kudos to @awni for working on this

I've added this info to readme

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants