Field Modules# TODO: High level description of field modules and how they connect together. Encodings Encoding Encoding.forward() HashEncoding HashEncoding.forward() HashEncoding.get_out_dim() HashEncoding.hash_fn() HashEncoding.pytorch_fwd() Identity Identity.forward() Identity.get_out_dim() NeRFEncoding NeRFEncoding.forward() NeRFEncoding.get_out_dim() RFFEncoding RFFEncoding.forward() RFFEncoding.get_out_dim() SHEncoding SHEncoding.forward() SHEncoding.get_out_dim() ScalingAndOffset ScalingAndOffset.forward() ScalingAndOffset.get_out_dim() TensorCPEncoding TensorCPEncoding.forward() TensorCPEncoding.get_out_dim() TensorCPEncoding.upsample_grid() TensorVMEncoding TensorVMEncoding.forward() TensorVMEncoding.get_out_dim() TensorVMEncoding.upsample_grid() Embeddings Embedding Embedding.build_nn_modules() Embedding.forward() Embedding.mean() Field Heads DensityFieldHead FieldHead FieldHead.forward() FieldHead.set_in_dim() FieldHeadNames PredNormalsFieldHead PredNormalsFieldHead.forward() RGBFieldHead SHFieldHead SemanticFieldHead TransientDensityFieldHead TransientRGBFieldHead UncertaintyFieldHead MLP MLP MLP.build_nn_modules() MLP.forward() Spatial Distortions SceneContraction SceneContraction.forward() SpatialDistortion SpatialDistortion.forward()