Merge "Add amm experiment." into main