All that being said though, I think there’s definitely potential. I would say give it a go if you have the time!
Would be very interested to hear any results or feedback
OK, well not a big surprise but it didn’t work - I got a version conflict with Tensorflow, followed by a crash (which I am assuming is related, but not necessarily).
Tensorflow Addons supports using Python ops for all Tensorflow versions above or equal to 2.2.0 and strictly below 2.4.0 (nightly versions are not supported).
The versions of TensorFlow you are currently using is 2.4.0-rc0 and is not supported.
Some things might work, some things might not.
Still, it’s hopeful that a big performance increase might come our way sometime - the benchmarks are with the new macs (m1 processor) but also using the GPU on a mac pro (intel) machine.
Thanks for trying that out @duncsand - and for posting your findings.
I also think that eventually we will be able to benefit from this.
Once this becomes more mature we will definitely look into it.