AML Assignment for learning potential