Gradient Descent on a Non-Convex Loss | TensorTonic