Apple M1 CPU + TensorFlow 2.4 + Rasa 2

RASA uses TensorFlow under the hood. TensorFlow 2.4 running on the recently-announced Apple M1 CPU has the potential to be significantly faster at training RASA models compared to all existing hardware[1]. Will this be coming to RASA Open Source in the near future? How can I track related development? Can I help make this happen?

[1] Accelerating TensorFlow Performance on Mac — The TensorFlow Blog

1 Like

From poking through the docs, it looks like it should already work? You’d just need the appropriate tf branch downloaded & running locally. You might need to update your Rasa tensorflow config as well.

Thanks Rachael. Unfortunately I can’t get Rasa to work on my M1 Mac (yet). When I try 2.2.4 (which uses Tensorflow 2.3.2) under Rosetta (x86 emulation) I get:

zsh: illegal hardware instruction rasa init

Looks like this has also been raised on StackOverflow:

Like Chris there, I also briefly attempted to use Apple’s alpha build of M1-optimized TensorFlow 2.4, but to no avail.

Any idea if anyone within Rasa is working on an M1? In the meantime I’ll revert to my older Intel Mac.

Looks like this is also being discussed here: Accelerated Tensorflow on mac - with no clear path forwards yet.

Seems like now Apple has released support for TensorFlow 2.5 on M1. Perhaps Rasa 2 can now move forward to have support on the M1 !