JIT-enabled LayerNormLSTM, it is much faster than the raw implementation !

Sync examples/mdn and examples/VAE to the latest API design