ferritin_onnx_models/
lib.rs

1//! Ferritin Onnx Models
2//!
3//! This crate provides easy access to various ONNX models for protein and ligand prediction.
4//! The models are downloaded from HuggingFace and run using ONNX Runtime.
5//! Currently supports ESM2 and LigandMPNN models.
6//!
7pub mod models;
8pub mod utilities;
9
10// pub use models::amplify::{AMPLIFYModels, AMPLIFY};
11pub use models::esm2::{ESM2, ESM2Models};
12pub use models::ligandmpnn::{LigandMPNN, ModelType};
13pub use utilities::{ndarray_to_tensor_f32, tensor_to_ndarray_f32, tensor_to_ndarray_i64};