Tensorflow is/was decent. It looked like they made a lot of effort for it to be accessible for outsiders.

Have you tried building the damn thing ?

Nix build is still stuck in the one from 3-4 y back because bazel doesn't play well. Debian too has some issues building the thing...