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;