mllf API
File Handling
For detailed documentation on file formats and usage examples, see File Handling.
RTF file parser |
|
PDB file parser |
|
Write bias coefficient files in the old .inp or .py style used by the simulator. |
|
Generate all combinations of site/sub files into separate directories. |
DeepSet Modules
For detailed documentation on DeepSet pretraining, see AtomBondGNN Pretraining.
DeepSet Autoencoder for pretraining atom-level embeddings. |
|
Dataset generation for DeepSet autoencoder pretraining. |
|
Training script for DeepSet autoencoder pretraining. |
|
Graph and CB Modules
For detailed documentation on CB architecture, see Contextual Bandit Setup.
Graph structure for contextual bandit. |
|
Edge value policy built on top of a node encoder. |
|
Value network for baseline estimation in REINFORCE. |
|
RGCN encoder using PyTorch Geometric. |
|
Helpers to convert existing Graph objects to PyTorch Geometric Data. |
Workflow Utilities
For detailed documentation on the workflow system, see Workflow System.
High-level workflow utilities for preparing combos, training and running sims. |