libmir-cuda 0.2.0

CUDA inference backend for libmir
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
mod attention;
mod clip;
mod elementwise;
mod patch;
mod pooling;
mod spatial_merge;
mod splice;

pub use attention::{VisionAttention, VisionAttentionSpec, VisionSpatialRope};
pub use clip::{VisionClip, VisionClipSpec};
pub use elementwise::{VisionElementwise, VisionElementwiseSpec};
pub use patch::VisionPatchLayout;
pub use pooling::{VisionPool, VisionPoolSpec};
pub use spatial_merge::SpatialMergeKernels;
pub use splice::VisionEmbeddingSplice;