--reset --mb=1 --dt=f4_e2m1:f4_e2m1:bf16 --dir=fwd_d --batch=shapes_4bit
--reset --mb=1 --dt=f16:f4_e2m1:f4_e2m1 --dir=bwd_d --batch=shapes_4bit
--reset --mb=1 --dt=f4_e2m1:f16:f4_e2m1 --dir=bwd_w --batch=shapes_4bit
--reset --mb=1 --dt=f4_e2m1 --dir=fwd_d --batch=shapes_4bit
