Molecular Representation & Property Prediction
Collection
Transformer encoders for SMILES/graph/image molecular representations — embeddings, ADMET & MoleculeNet property prediction. • 3 items • Updated
How to use Aurigene-AI/ChemBERTa-77M-MTR with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, RobertaForRegression
tokenizer = AutoTokenizer.from_pretrained("Aurigene-AI/ChemBERTa-77M-MTR")
model = RobertaForRegression.from_pretrained("Aurigene-AI/ChemBERTa-77M-MTR", device_map="auto")Mirrored by Aurigene AI
Discovery stage: Lead optimization
ChemBERTa-2 (77M) pretrained with multitask regression over 200 normalised RDKit descriptors. A strong, cheap baseline for MoleculeNet-style ADMET endpoints.
Upstream:
DeepChem/ChemBERTa-77M-MTR- all credit to the original authors; the model card and licence below are theirs.Explore the rest of the catalogue: Molecule Explorer - Protein Target Explorer - Drug Discovery Model Hub
See the upstream model card for full details, training data and citation.