Function Java_com_torsh_nn_Linear_nativeCreateLinear 

#[unsafe(no_mangle)]
pub unsafe extern "C" fn Java_com_torsh_nn_Linear_nativeCreateLinear(
    _env: JNIEnv,
    _class: jclass,
    in_features: jint,
    out_features: jint,
    bias: jboolean,
) -> jlong

JNI function to create a linear layer