-record(linear_layer, {
    weight :: viva_tensor@native@cuda:accelerated_tensor(),
    bias :: viva_tensor@native@cuda:accelerated_tensor(),
    input_features :: integer(),
    output_features :: integer(),
    backend :: viva_tensor@native@cuda:acceleration_backend()
}).