Solved by SimLeek in post #6 
              
              
                Alright, I’ve got this fully implemented here: https://github.com/SimLeek/TorchRender/blob/master/torchrender/pixel_shader.py 
Now I just need to get tests on it and upload to pypi. 
For some reason I don’t completely understand, this setup works quite well: 
self.min_bounds = [0 for _ in frame.shap…