Mixed precision training support

[ogencoglu]

Does compute_gramian() support mixed precision training? I couldn't make it work at least.

[PierreQuinton]

Thanks for the issue!

It is not clear to me what the problem/goal is. Could you provide us with a minimal example of what you want to do?