I'm not certain how many other researchers would be willing to swap away from PyTorch (or tf or jax) to yet another framework. Amazon pushed MXNet for a while,

I'm not certain how many other researchers would be willing to swap away from PyTorch (or tf or jax) to yet another framework. Amazon pushed MXNet for a while, but the only time I used it was while interning there. From what I understand, advantage here is tight integration with Apple hardware/shared memory? That being said, going to spend some time getting familiar with it by reimplementing Llama-2 and trying to make it fast: https://github.com/jbarrow/mlxllama https://github.com/jbarrow/mlxllama