|
Mila
Deep Neural Network Library
|
import Compute.CudaSoftmaxOp;import Compute.CudaResidualOp;import Compute.CudaMatMulBiasGeluOp;import Compute.CudaLinearOp;import Compute.CudaMHAOp;import Compute.CudaLayerNormOp;import Compute.CudaGeluOp;import Compute.CudaEncoderOp;