Keras training with custom cost function: Inconsistent behavior

Question

I get different results during training when I try to implement a custom training loop and when I use the built in train on batch function. In particular, if I print it, I s