libmir-cuda 0.3.0

CUDA inference backend for libmir
1
2
3
4
5
6
7
8
9
mod embedding;
mod output;
#[cfg(all(test, target_os = "linux"))]
mod tests;

pub(in crate::backend::model) use embedding::{ModelEmbedding, ModelEmbeddingTemplate};
pub(in crate::backend::model) use output::{
    ModelBatchOutputHead, ModelOutputHead, ModelOutputHeadTemplate,
};