Can AI calculate the gravitational potential energy of an object?