I think you may be in the wrong place— you’re asking about TensorFlow in Python, right? If so, you should probably check out Stack Overflow. If you’re not talking about TensorFlow in Python, TensorFlow.js integration in Snap! is something that @toontalk made. This discussion about it includes loss becoming NaN.