You can now enable Gravity Forms plugin support in AMP with just one click. This will work with the form builder and all kinds of fields with proper form validation. Please note that this plugin need Gravity Forms plugin to be installed in your website.
Please make sure you have installed AMP for WP plugin, before installing this add-on.