Awni Hannun
Awni Hannun
Verified this fixes cifar example + Batch Norm
From a design / perf / testing standpoint I think this is all exactly right and beautifully done. Let's PR it. I want to think a bit more about removing...
Closes https://github.com/ml-explore/mlx-examples/issues/233
What version of MLX are you using? Building from source, if so what commit? I have to admit I've never seen anything even close to that message you are showing....
> the kernel here refers to the IPython or Jupyter kernel Makes a lot more sense, thanks!
Cool, we follow the NumPy API so the `where` and `unique` are definitely on the table for feature enhancements. They are both tricky to expose in MLX since they produce...
> The code is not compiling for some reason when pip install -e . is run 🤔 Were you able to get it to build using Cmake directly?
I'm not able to repro the build failure. It build fine for me 😅 ... maybe try clearing out your build caches..
Does it build for you now?
Yup we should add that and `logical_or` and the overloads. It shouldn't be too difficult to get those in.