Jiarui Fang 16cc8e6aa7 [builder] MOE builder (#2277)

2023-01-03 20:29:39 +08:00

..

__init__.py

[builder] MOE builder (#2277)

2023-01-03 20:29:39 +08:00

builder.py

[builder] MOE builder (#2277)

2023-01-03 20:29:39 +08:00

cpu_adam.py

[builder] polish builder with better base class (#2216)

2022-12-28 19:45:49 +08:00

fused_optim.py

[builder] polish builder with better base class (#2216)

2022-12-28 19:45:49 +08:00

moe.py

[builder] MOE builder (#2277)

2023-01-03 20:29:39 +08:00

multi_head_attn.py

[builder] polish builder with better base class (#2216)

2022-12-28 19:45:49 +08:00

scaled_upper_triang_masked_softmax.py

[builder] builder for scaled_upper_triang_masked_softmax (#2234)

2022-12-30 09:58:00 +08:00

utils.py

[buider] use builder() for cpu adam and fused optim in setup.py (#2187)

2022-12-23 16:05:13 +08:00