Skip to content

Do you guys want the Python nn and optimizer modules written in C++? #774

Answered by awni
cemlyn007 asked this question in Q&A
Discussion options

You must be logged in to vote

It's something we've discussed adding in the past, but haven't quite found enough of a reason to do it. At the moment we're not planning to add that level in C++ until there is enough demand for it (as it adds another surface for us to maintain).

FWIW we also have MLX Swift which has all the high-level layers + otpimizers: https://github.com/ml-explore/mlx-swift/. I'm not sure what your goals are but maybe that would be a good alternative.

However, if this is something super useful for you, I would encourage you to start a separate project which builds some of those layers out on top of MLX in C++. That would make it 1. more concrete so we can see how it looks and if it makes sense to inc…

Replies: 3 comments 1 reply

Comment options

You must be logged in to vote
0 replies
Comment options

You must be logged in to vote
1 reply
@cemlyn007
Comment options

Answer selected by cemlyn007
Comment options

You must be logged in to vote
0 replies
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
3 participants