Gradient_transformers was removed from Tensorflow rendering the algorithm unusable. Is there any alternative ?