Function Java_com_torsh_optim_Adam_nativeCreateAdam 

#[unsafe(no_mangle)]
pub unsafe extern "C" fn Java_com_torsh_optim_Adam_nativeCreateAdam(
    _env: JNIEnv,
    _class: jclass,
    learning_rate: jfloat,
    beta1: jfloat,
    beta2: jfloat,
    epsilon: jfloat,
) -> jlong

JNI function to create Adam optimizer