def forward(self, xs):
    """Executes this layer as part of a forward pass through the model.

    Args:
      xs: Unused tensors.

    Returns:
      Random uniform tensor of the shape and type specified in constructor.
    """
    rng = self._get_conditionally_synced_rng()
    result = fastmath.random.uniform(
        rng, self._shape, self._dtype, self._min_val, self._max_val)
    return result
Executes this layer as part of a forward pass through the model.

    Args:
      xs: Unused tensors.

    Returns:
      Random uniform tensor of the shape and type specified in constructor.
    
trax/layers/core.py