For the reference, this is part of the experiments we are currently doing to try out different potential backend in https://github.com/pymc-devs/pymc4_prototypes
The previous discussions could be seen in https://github.com/pymc-devs/pymc4_prototypes/pull/4