Python implementation of Torso, Policy head and Value head

[singh-is-singh]

Any pointers to python implementation of Torso, Policy head and Value head ?

Essentially, the python code for below figures:

https://www.nature.com/articles/s41586-022-05172-4/figures/8
https://www.nature.com/articles/s41586-022-05172-4/figures/9