colossal_C
colossal_layer_norm_cuda
colossal_multihead_attention
colossal_scaled_masked_softmax
colossal_scaled_upper_triang_masked_softmax
colossalai
model_zoo
