I am not sure to understand why Keras models don’t have this issue given that they are supposed to implement exactly the same thing. What is the implementation difference that makes Keras not have this behaviour?