Become a MacRumors Supporter for $50/year with no ads, ability to filter front page stories, and private forums.

jerryk

macrumors 604
Nov 3, 2011
7,418
4,207
SF Bay Area
They certainly won't run anything but toy models on their Macs. These are desktop systems to work on and connect to larger clusters. A Macbook Air would be totally fine for that type of work and I doubt they have availability issues with Nvidia hardware.
Yes. Most people I know do a lot of work of their work in the cloud running in data centers and use big datasets, GPU/TPU processing power, and lots of drive space to build reasonably sized models.
 

senttoschool

macrumors 68030
Nov 2, 2017
2,575
5,338
Yes. Most people I know do a lot of work of their work in the cloud running in data centers and use big datasets, GPU/TPU processing power, and lots of drive space to build reasonably sized models.
Even further, a lot of tech companies here in San Francisco / SV are switching to cloud IDEs and cloud dev environments. So the computer is just a shell. Macs are still used though.

At my tech company, we're piloting this now with a few devs.
 
  • Like
Reactions: jerryk

Xiao_Xi

macrumors 68000
Oct 27, 2021
1,529
955
This looks cool.
MLX is an array framework for machine learning on Apple silicon, brought to you by Apple machine learning research.
 

GrumpyCoder

macrumors 68020
Nov 15, 2016
2,074
2,654
That's certainly very interesting, but I have to wonder how much support it will see in the future. First Apple got involved in TF, then they pretty much abandoned it to get involved with PyTorch a while later, which looks like they forgot about as well. And now this is the next thing for them... the big question is for how long until it's abandoned?
Wish they would stick to something, fix it when issues are discovered and generally support it well.

I'll keep an eye on this. I just had to get rid of some money from this years budget, so I ordered some more Mac Studios. So maybe I'll setup a small Mac MLX playground and see how well this works.
 
  • Like
Reactions: leman

name99

macrumors 68020
Jun 21, 2004
2,282
2,139
That's certainly very interesting, but I have to wonder how much support it will see in the future. First Apple got involved in TF, then they pretty much abandoned it to get involved with PyTorch a while later, which looks like they forgot about as well. And now this is the next thing for them... the big question is for how long until it's abandoned?
Wish they would stick to something, fix it when issues are discovered and generally support it well.

I'll keep an eye on this. I just had to get rid of some money from this years budget, so I ordered some more Mac Studios. So maybe I'll setup a small Mac MLX playground and see how well this works.
I think that's to misinterpret things.

Apple worked on TF and PyTorch because in both cases there was obvious value in writing the "drivers" (ie the glue code) that would route TF and PyTorch primitives to appropriate Apple hardware, whether GPU or ANE. Once that was done, there was no need for Apple to stick around; their job is not to tell TF or PyTorch how to write new API's.

Meanwhile MLX is maybe positioned somewhat in the same place as Metal, namely a set of preferred API's and tools that are optimized (and will remain optimized) for Apple HW. The host for all ML code that Apple writes internally, and the suggested host for all external code where people are willing to do some extra work to get optimal Apple performance?
 

Xiao_Xi

macrumors 68000
Oct 27, 2021
1,529
955
Apple worked on TF and PyTorch because in both cases there was obvious value in writing the "drivers" (ie the glue code) that would route TF and PyTorch primitives to appropriate Apple hardware, whether GPU or ANE. Once that was done, there was no need for Apple to stick around; their job is not to tell TF or PyTorch how to write new API's.
The Metal backend for Tensorflow and Pytorch is still in beta, so Apple still has work to do. On top of that, I doubt that Tensorflow and Pytorch developers will support the Metal backend, so Apple should fix any issues with those frameworks on Apple hardware.

Wish they would stick to something, fix it when issues are discovered and generally support it well.
I hope Apple would create an MLX-based backend at least for Pytorch.
 

leman

macrumors Core
Oct 14, 2008
19,319
19,336
That's certainly very interesting, but I have to wonder how much support it will see in the future. First Apple got involved in TF, then they pretty much abandoned it to get involved with PyTorch a while later, which looks like they forgot about as well. And now this is the next thing for them... the big question is for how long until it's abandoned?
Wish they would stick to something, fix it when issues are discovered and generally support it well.

I'll keep an eye on this. I just had to get rid of some money from this years budget, so I ordered some more Mac Studios. So maybe I'll setup a small Mac MLX playground and see how well this works.

I agree. So far, Apples attempts to bring ML development to Apple Silicon is full of half-baked efforts. They have released a series of incomplete buggy backends only seemingly to abandon them very soon thereafter. Same goes for their fairly new ML Compute framework which is now marked as deprecated.

MLX also seems incomplete, and the documentation is very minimal. I do appreciate the fact that it is MIT licensed and has a C++ implementation. Maybe the idea is to use MLX as a new backend for the future TF/PyTorch/Jax compatibility? I’m a bit concerned that they are rolling their own Python API, surely they don’t consider challenging the big frameworks? I don’t see a world where it would be wise.


I think that's to misinterpret things.

Apple worked on TF and PyTorch because in both cases there was obvious value in writing the "drivers" (ie the glue code) that would route TF and PyTorch primitives to appropriate Apple hardware, whether GPU or ANE. Once that was done, there was no need for Apple to stick around; their job is not to tell TF or PyTorch how to write new API's.

The problem is that these drivers are incomplete, buggy, badly documented, and closed source - at least for TF, not sure about others). So the job is hardly done. MLX is just the last preview-quality implementation in a series of related efforts, and Apples track record here is not the most reliable.

Meanwhile MLX is maybe positioned somewhat in the same place as Metal, namely a set of preferred API's and tools that are optimized (and will remain optimized) for Apple HW. The host for all ML code that Apple writes internally, and the suggested host for all external code where people are willing to do some extra work to get optimal Apple performance?

True, and there is certainly value in having a high quality abstract layer that can be used to quickly write framework drivers. But as I wrote I am a bit concerned that they are implementing their own JAX clone, it’s very easy to get lost in an effort like that and I don’t see much chance for MLX to be widely adopted.
 
  • Like
Reactions: Xiao_Xi

GrumpyCoder

macrumors 68020
Nov 15, 2016
2,074
2,654
Apple worked on TF and PyTorch because in both cases there was obvious value in writing the "drivers" (ie the glue code) that would route TF and PyTorch primitives to appropriate Apple hardware, whether GPU or ANE. Once that was done, there was no need for Apple to stick around; their job is not to tell TF or PyTorch how to write new API's.
In general I would agree, but what Apple delivered so far doesn't work as expected. It's bugged and produces non-reliable results. We can't really use it other than for playing around and try a few things. But when the results can be different from running it on a CPU (on any platform including Apple) or other GPU platforms, then it's really of no use. I'm not sure if that's fixed now, as everyone I know abandoned it as well after waiting for bug fixes. That might be the reason why Apple never fixed it though, everyone left already.
MLX also seems incomplete, and the documentation is very minimal. I do appreciate the fact that it is MIT licensed and has a C++ implementation. Maybe the idea is to use MLX as a new backend for the future TF/PyTorch/Jax compatibility? I’m a bit concerned that they are rolling their own Python API, surely they don’t consider challenging the big frameworks? I don’t see a world where it would be wise.
No doubt it's incomplete. I'm fine with that for now though. I just hope they have not brought it into existence and are done now. A monthly patch with patch notes (what's been fixed, what new features added) would be nice to have from now on. And no, I can't imagine they want to replace TF/PyTorch for everyone. That's not going to happen.
 

leman

macrumors Core
Oct 14, 2008
19,319
19,336
No doubt it's incomplete. I'm fine with that for now though. I just hope they have not brought it into existence and are done now. A monthly patch with patch notes (what's been fixed, what new features added) would be nice to have from now on. And no, I can't imagine they want to replace TF/PyTorch for everyone. That's not going to happen.

They do seem more passionate about this particular project, with devs actively answering questions on twitter and hacker news. Of course, only convenient questions ^^ We will see how it goes.

For me it's also very interesting as I am currently developing a library for basic GPU algorithms (e.g. sort), and it would be a nice test benchmarking it against what Apple engineers wrote.
 

name99

macrumors 68020
Jun 21, 2004
2,282
2,139
In general I would agree, but what Apple delivered so far doesn't work as expected. It's bugged and produces non-reliable results. We can't really use it other than for playing around and try a few things. But when the results can be different from running it on a CPU (on any platform including Apple) or other GPU platforms, then it's really of no use. I'm not sure if that's fixed now, as everyone I know abandoned it as well after waiting for bug fixes. That might be the reason why Apple never fixed it though, everyone left already.

No doubt it's incomplete. I'm fine with that for now though. I just hope they have not brought it into existence and are done now. A monthly patch with patch notes (what's been fixed, what new features added) would be nice to have from now on. And no, I can't imagine they want to replace TF/PyTorch for everyone. That's not going to happen.

Is the real conclusion that there's so much flux in this space that Apple, being Apple, simply aren't going to waste time continuing to support a model (TF and PyTorch) that they think is not very good? Same way they spent a few years with OpenGL and OpenCL, learned all the things that made these suck, and walked away from them?

I honestly don't know enough about this space to have opinions. I do know that there are always people wedded to old code who refuse to move to something new (hell, there are still people writing GL) and they tend to have very loud voices; but that doesn't change the fact that Apple has always been a company that will drop a technology (HW or SW) as soon as it thinks it has discovered something better.

Another possibility is that they believe they have laid out the path for how to improve TF and PyTorch on Apple Silicon, and now it's up to that open source community, that's *always* boasting about how fantastic it is, to implement the last few steps, now that Apple has made it clear what needs to be done?

One final question. Where does Mojo fit in all this? If Mojo is going to be the optimal way to execute Python/PyTorch going forward (everywhere, but especially on Apple Silicon) once again why would Apple bother continuing to support a sub-optimal solution?

This might sound confrontational, but it isn't meant to be! I genuinely do not know the answers to these questions; they're just issues that arise to me as aspects of what might be driving Apple's choices.
 

Xiao_Xi

macrumors 68000
Oct 27, 2021
1,529
955
Another possibility is that they believe they have laid out the path for how to improve TF and PyTorch on Apple Silicon, and now it's up to that open source community
I can't imagine that Apple would be naive enough to think that major backers of Tensorflow and Pytorch, such as Google, Intel or Nvidia, will help optimize either of those frameworks on Apple hardware.
 
  • Like
Reactions: flybass

flybass

macrumors member
May 1, 2015
85
93
I can't imagine that Apple would be naive enough to think that major backers of Tensorflow and Pytorch, such as Google, Intel or Nvidia, will help optimize either of those frameworks on Apple hardware.
Yep. To be blunt, the reality here is that AI at Apple is a joke. Also, no serious AI researchers are doing experiments on Mac hardware.
 

altaic

macrumors 6502a
Jan 26, 2004
657
436
I can't imagine that Apple would be naive enough to think that major backers of Tensorflow and Pytorch, such as Google, Intel or Nvidia, will help optimize either of those frameworks on Apple hardware.
At the most basic level, the Metal kernels from mlx are useful to PyTorch etc. Getting a bit more in depth, the library architecture will inform and give insight WRT hardware and applications. As I see it, it’s a win-win.
 

Xiao_Xi

macrumors 68000
Oct 27, 2021
1,529
955
Some people have already started playing with MLX.

Edit: The author of the post has added a RTX 4090 result using a more optimized implementation of Whisper for Nvidia GPUs.

PerfChart.jpg
 
Last edited:

racetrack

macrumors newbie
Jan 2, 2023
2
1
I do prototyping in MATLAB before things over to PyTorch, and I know others (mostly engineering folk) who do the same. My current project involves piping high resolution signals through two LSTMs and a 1D CNN. It would likely take months from start to finish on a CPU. Unfortunately only Nvidia GPUs are usable in MATLAB, and PyTorch support and performance for other vendors is grim.

I can manage VRAM issues with smaller batches and other tricks (fitting my current project into the 16GB VRAM I have available is manageable), but I can't manage at all without GPU support. Until Apple improves PyTorch support and performance, they could offer 1TB VRAM for all I care. My work is impossible using CPU cores (except for inference steps within the pipeline which are actually a tad faster). A real shame.
 
Last edited:
  • Like
Reactions: Chuckeee

Xiao_Xi

macrumors 68000
Oct 27, 2021
1,529
955
I find it interesting that Apple is promoting Swift and MLX for ML researchers.
 
  • Like
Reactions: Flowstates
Register on MacRumors! This sidebar will go away, and you'll see fewer ads.