DeepSpeed

4abd4555 - add batch-size at the tranform launch for the half-precision implementation