See also https://github.com/unifyai/ivy https://github.com/unifyai/ivy which I have not tried but seems along the lines of what you are describing, working with

See also https://github.com/unifyai/ivy https://github.com/unifyai/ivy which I have not tried but seems along the lines of what you are describing, working with all the major frameworks