The tool is written by Daniel Smilkov and Shan Carter from Google Brain's team.
It is a great tool to understand using small examples the network operation.
The tool promised:
It took me 5 minutes to find a configuration which breaks it ! :-)
Here it is:
Note that both Test loss and training loss are NaN. I am waiting for the fix. (I see this error was already reported)