bytedance / lightseq

LightSeq: A High Performance Library for Sequence Processing and Generation
Other
3.22k stars 329 forks source link

support ViT model #299

Closed zjersey closed 2 years ago

zjersey commented 2 years ago

support training and inference of Vision Transformer (ViT), provide examples; remove useless moe kernel;