tribev2 0.0.4

TRIBE v2 — multimodal fMRI brain encoding model inference in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

pub mod scalenorm;
pub mod rotary;
pub mod attention;
pub mod feedforward;
pub mod residual;
pub mod encoder;
pub mod projector;
pub mod subject_layers;
pub mod tribe;
pub mod weights;

#[cfg(feature = "wgpu-kernels-metal")]
pub mod kernels;
#[cfg(feature = "wgpu-kernels-metal")]
pub mod fused;

#[cfg(feature = "wgpu-kernels-metal")]
pub use fused::FusedOps;