Keras 2 compatibility

[zzaibi]

I struggled a lot to make the code under keras 2. Could you provide an updated version?

Thanks much!

[codekansas]

Hi, I don't plan to update this repo to Keras 2. However, I've rewritten some of the classes in another repository using Keras 2 here, such as the recurrent attention wrapper.