pytorch3d/tutorials/fit_simple_neural_radiance_field