cs330 (Structural Grokking Project)