(forward and backward), adapted from Apex's[FusedDense](https://github.com/NVIDIA/apex/tree/master/apex/fused_dense).We make it work for bfloat16.For best performance, you should use CUDA >= 11.8. CuBLAS versions beforethis doesn't have the best matmul + bias + gelu performance for bfloat16.```shcd csrc/fused_dense_lib && pip install .```