Crate brainjepa 

brainjepa-rs — Brain-JEPA fMRI Foundation Model inference in Rust

Inference uses RLX (rlx-engine, default).

Binary	Purpose
infer	Encoder embeddings
classify	Downstream classification
predict	JEPA masked prediction

Re-exports

pub use config::DataConfig;

pub use config::ModelConfig;

pub use config::YamlConfig;

pub use csv_export::save_embeddings_csv;

pub use data::FmriInputF32;

pub use data::GradientData;

pub use error::BrainJepaError;

pub use error::Result;

pub use hf_download::resolve as resolve_weights;

pub use hf_download::ResolvedWeights;

pub use hf_download::DEFAULT_REPO;

pub use masks::full_context_mask;

pub use masks::jepa_masks;

pub use masks::mask_config_for;

pub use masks::random_block_mask;

pub use masks::MaskConfig;

pub use rlx::predict_class;

pub use rlx::AttnLayout;

pub use rlx::BrainJepaEncoder;

pub use rlx::BrainJepaPredictor;

pub use rlx::EmbeddingResult;

pub use rlx::RlxClassificationHead as ClassificationHead;

Modules

config

csv_export

CSV export utilities for Brain-JEPA embeddings.

data

error

hf_download

masks

Spatiotemporal masking for Brain-JEPA (engine-agnostic patch indices).

prelude

Convenience re-exports for common usage.

rlx

RLX-backed Brain-JEPA inference (rlx::Graph + rlx::Session).

Functions

init_threads

Configure the global Rayon thread pool.