esm

Model type	esm
Class	EsmModel
Task	feature-extraction
Source	models/esm.py

Description

ESM-2 protein encoder for per-residue feature extraction.

Replicates HuggingFace’s EsmModel; the output is last_hidden_state (the pooler and the contact head are not part of the embedding contract and are dropped).

Usage

mobius build --model <MODEL_ID> output_dir/

from mobius import build

model = build("<MODEL_ID>")