This is a simple Neural Network made with layers api from Tensorflow.js which is really cool library.
You can tweak the hyper parameters from the side bar.
Current learning rate = 0.5 , epoch=100 , itterations=100 (mobile phones may take some time for computing , and may show unidentified loss , please wait for some time to complete the computation)