add NeRF training

[jenkspt]

• Adapt jax code from sarafridov/plenoxels
  • Replace the plenoxel sparse grid with HashArray of
  • Replace spherical harmonic eval_sh with MLP

The instant neural graphics nerf training is more similar to plenoxel than original nerf, since plenoxel uses trilinear interpolation on a sparse grid.