# https://github.com/ridgerchu/matmulfreellm

import torch
import torch.nn.functional as F

from mmfreelm.models import HGRNBitConfig, HGRNBitModel