4 Commits (d10ee42f68d090db17a8b87cac46ab6d1c2c8ca2)

Author SHA1 Message Date
Hongxin Liu e5ce4c8ea6
[npu] add npu support for gemini and zero (#5067) 1 year ago
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752) 1 year ago
Frank Lee 40d376c566
[setup] support pre-build and jit-build of cuda kernels (#2374) 2 years ago
Frank Lee 9b765e7a69
[setup] removed the build dependency on colossalai (#2307) 2 years ago
Jiarui Fang 16cc8e6aa7
[builder] MOE builder (#2277) 2 years ago
Jiarui Fang db4cbdc7fb
[builder] builder for scaled_upper_triang_masked_softmax (#2234) 2 years ago
Jiarui Fang 1cb532ffec
[builder] multihead attn runtime building (#2203) 2 years ago
Jiarui Fang d42afd30f8
[builder] runtime adam and fused_optim builder (#2184) 2 years ago