Neural Scene Flow Fields using pytorch-lightning, with potential improvements
Use this model to run the notebook