std1.0.0-beta-7 SafeC docs

std :: ml::gpu_cuda

Function std::cuda_mp_adam

#include <std/ml/gpu_cuda.h> source ยท definition

int cuda_mp_adam(
    struct CudaBuf w,
    struct CudaBuf m,
    struct CudaBuf v,
    struct CudaBuf g,
    struct CudaHalfBuf wHalf,
    float lr,
    float beta1,
    float beta2,
    float eps,
    float weightDecay,
    int decoupled,
    unsigned long step,
    struct CudaBuf skip,
)