Expand description
ESM3 VQ-VAE structure token encoder and decoder (vqvae.py port).
StructureTokenEncoder maps backbone coordinates to discrete structure tokens via a
2-layer geometric mini-transformer + nearest-neighbour VQ codebook lookup.
StructureTokenDecoder is stubbed (not needed for ESM3 inference).
Structsยง
- Structure
Token Decoder - Stub for the structure token decoder. Not needed for ESM3 inference.
- Structure
Token Encoder - Encodes backbone coordinates into discrete structure tokens via a 2-layer geometric mini-transformer followed by nearest-neighbour VQ codebook lookup.
- VqVae
Config