Tensorflow example works fine on colab, but gives nan while running on jupyter lab

Question

After working with the tensorflow\'s example tutorial on colab: Basic regression: Predict fuel efficiency, which can be found here: https://www.tensorflow.org/tutorials/kera