We already have support for FP32. Let's expand the coverage to these dtypes. cc @larryliu0820 @manuelcandales