This is great, and the paper+codebase they're referring to (but not linking, here [1]) is neat too.
The research is moving fast though, so if you want something almost as fast without specialized CUDA kernels (just plain pytorch) you're in luck: https://github.com/apchenstu/TensoRF
As a bonus you also get a more compact representation of the scene.