Crate ferritin_onnx_models

Source
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;

Modules§

models
utilities