v0.3.1
[bf16] add bf16 support (#3882)
fix Tensor is not defined (#4129)
[Kernels] added triton-implemented of self attention for colossal-ai (#4241)
[setup] support pre-build and jit-build of cuda kernels (#2374)