[FEA] BSDG: Support `fit` on dtypes other than `fp32`

[jinsolp]

BSDG (Billion Scale Data Generator)

Current implementation only supports fit on fp32 dtypes.
Add support for other types (e.g. bf16, int8 etc)