import torch

from ._ops import ops

from . import layers

def rmsnorm_forward(x: torch.Tensor, weight: torch.Tensor) -> torch.Tensor:
    return ops.rmsnorm_forward(x, weight)

__all__ = [
    "layers",
    "rmsnorm_forward",
]