main.py

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'  # stfu tensorflow

import tensorflow as tf

x = tf.constant(3.0)
print(x)

with tf.GradientTape() as tape:
    tape.watch(x)
    y = x * x

dy_dx = tape.gradient(y, x)
print(dy_dx)

Suppress tensorflow warnings and example usage of tf.GradienTape