Pytorch implementation of tiny NeRF model - View it on GitHub