Refactor config/code classes to follow Flax.

[Łukasz Lew]

For every piece of logic (like Numerics, Calibration, Tensor, DotGeneralRaw, DotGeneral, we should have a single class with "dataclass" field that configure it and methods that execute the logic.