Build and train neural network models using TensorFlow 2.x