Expand description
Ferritin Onnx Models
This crate provides easy access to various ONNX models for protein and ligand prediction. The models are downloaded from HuggingFace and run using ONNX Runtime. Currently supports ESM2 and LigandMPNN models.
Re-exports§
pub use models::esm2::ESM2;
pub use models::esm2::ESM2Models;
pub use models::ligandmpnn::LigandMPNN;
pub use models::ligandmpnn::ModelType;
pub use utilities::ndarray_to_tensor_f32;
pub use utilities::tensor_to_ndarray_f32;
pub use utilities::tensor_to_ndarray_i64;