├── README.md └── lightseq ├── async_communication.py ├── lightseq_async_attn.py ├── lightseq_async_attn_varlen.py ├── lightseq_ckpt_monkey_patch.py └── train_lightseq_no_trainer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RulinShao/LightSeq/HEAD/README.md -------------------------------------------------------------------------------- /lightseq/async_communication.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RulinShao/LightSeq/HEAD/lightseq/async_communication.py -------------------------------------------------------------------------------- /lightseq/lightseq_async_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RulinShao/LightSeq/HEAD/lightseq/lightseq_async_attn.py -------------------------------------------------------------------------------- /lightseq/lightseq_async_attn_varlen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RulinShao/LightSeq/HEAD/lightseq/lightseq_async_attn_varlen.py -------------------------------------------------------------------------------- /lightseq/lightseq_ckpt_monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RulinShao/LightSeq/HEAD/lightseq/lightseq_ckpt_monkey_patch.py -------------------------------------------------------------------------------- /lightseq/train_lightseq_no_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RulinShao/LightSeq/HEAD/lightseq/train_lightseq_no_trainer.py --------------------------------------------------------------------------------