TensorFlow.js Performance Tuning