Based on https://github.com/ashawkey/torch-ngp - View it on GitHub