Stop_gradient/detach equivalent in flux

[Rassibassi]

Great! Thanks a lot, it made my code work example over here: Autoencoder for telecommunication (Constellation shaping)