neural_net_physics_tf backend model training code for physics in the browser project, TensorFlow version