Bf16 might be freezing your weight updates | HACKOBAR_