fabric = Fabric(accelerator="cuda", precision="bf16-mixed")