replace pwa with differentiable version
Merge pull request #8 from nontas/atm_transform_fix

bugfix in ATMs