Faster-WAM in LeRobot: from 300 cache reads to 10
Faster-WAM cuts a world action model's action expert to one layer. I added it to LeRobot behind opt-in flags and timed it on an A100 with random weights.
world-action-modelslerobotkv-cacheflow-matching