brainharmony 0.1.0

Brain-Harmony multimodal brain foundation model — inference in Rust with Burn ML
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

/// Convenience re-exports for common usage.
///
/// ```rust,ignore
/// use brainharmony::prelude::*;
/// ```
pub use crate::classification::{ClassificationHead, MLPHead, predict_classes};
pub use crate::config::{DataConfig, ModelConfig, YamlConfig};
pub use crate::csv_export::save_embeddings_csv;
pub use crate::data::{GradientData, GeohData, SignalInput};
pub use crate::error::{BrainHarmonyError, Result};
pub use crate::hf_download::{resolve as resolve_weights, ResolvedWeights, DEFAULT_REPO};
pub use crate::inference::{BrainHarmonyEncoder, EmbeddingResult};
pub use crate::predictor_api::BrainHarmonyPredictor;
pub use crate::masks::{full_context_mask, jepa_masks, MaskConfig};
pub use crate::weights::{WeightFilter, WeightMap};