Significant performance loss of ML code on GPU

[dkobayas-cyber]

Here are the posts I referred to:

• Same model running on GPU and CPU produce different results
• Different loss for cpu and gpu