switch transformers

A model introduced by Google Research, it uses a mixture of experts approach to scale up the model size without significantly increasing computation requirements.