V2

[CurtinComputing]

Hi developers,

I find there are still a lot of issues caused by the code bugs on the Nerfies@v2.
These issues has been opened on the issue sections inclduing #58 #59 #53 #44 .

The main correct on the bugs of code is as listed as below:

1. Under coding file "configs.py", at line 19 "from flax import nn" should be corrected to "from flax import linen as nn"
2. Under coding file "model_utils.py", at line 107 " jnp.broadcast_to([last_sample_z], z_vals[..., :1].shape)" should be corrected to " jnp.broadcast_to(jnp.array([last_sample_z]), z_vals[..., :1].shape)"

After fixing these bugs of the codes, nerfies can run the with the prepaired custmised image data.

Larry