import numpy as np def lora_forward(x, W, b, A, B, alpha): raise NotImplementedError def merge_lora(W, A, B, alpha): raise NotImplementedError def count_trainable(in_features, out_features, rank): raise NotImplementedError