Render rays Args: H: int. Height of image in pixels. W: int. Width of image in pixels. focal: float. Focal length of pinhole

Transforms model's predictions to semantically meaningful values. Args: raw: [num_rays, num_samples along ray, 4]. Prediction from model

Constructs a version of 'fn' that applies to smaller batches.

Render rays in smaller minibatches to avoid OOM.

Instantiate NeRF's MLP model.

Volumetric rendering. Args: ray_batch: array of shape [batch_size, ...]. All information necessary for sampling along a ray, incl

Prepares inputs and applies network 'fn'.