Hi @osterhult, sorry for the delay, I have taken this on and am working on a branch, but among other things I had some hurdles updating to Big Sur (without which I could not install tensorflow_macos). I will update this when I have more information.