The poor performance here isn't due to TensorFlow.js itself. It has to do with how I'm updating the DOM with the colors as the network is being trained. I could

The poor performance here isn't due to TensorFlow.js itself. It has to do with how I'm updating the DOM with the colors as the network is being trained. I could make a few tweaks here to get a dramatic improvement in performance. You are also correct that it would be even less resource intensive if we just served the model instead of training it in the browser. You should try what you are thinking of doing. I think you will be pleasantly surprised at how well it does.