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.

Sign up for GitHub

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

Jump to bottom

MLX updates #12

Open
altaic opened this issue Aug 8, 2024 · 1 comment
Open

MLX updates #12

altaic opened this issue Aug 8, 2024 · 1 comment

Comments

@altaic
Copy link

altaic commented Aug 8, 2024 •
edited
Loading

FYI, einsum just landed in mlx along with python array api support. Also, thanks for einx!

ml-explore/mlx#1269
ml-explore/mlx#1289
@fferflo
Copy link
Owner

fferflo commented Aug 8, 2024

Hi, thanks for bringing this up! Support for mx.einsum (einx.dot) is already implemented, although not in a release version yet: 0ad7af3 It will be included in the next release of einx. (Using some other einx functions such as einx.get_at with mlx is currently limited to the use cases that are supported by mx.vmap)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@altaic @fferflo