https://analyticsindiamag.com/a-deep-dive-into-switch-transformer-architecture/