import torch


def make_A(
    As, ns, device=torch.device("cpu"), dtype=torch.float32
):  # pylint: disable=unused-argument
    """Move the As to the provided device, convert to the provided dtype, and
    return as A."""
    As = [A_j.to(device, dtype) for A_j in As]
    return As