Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
MolCraftDiffusion
Logo

Getting Started

  • Installation

Tutorials

  • Tutorial 0: Data Preparation & Management
  • Tutorial 1: Training a Diffusion Model
  • Tutorial 2: Training a Regressor Model
  • Tutorial 3: Training a Time-Aware Guidance Model
  • Tutorial 4: Fine-Tuning a Diffusion Model
  • Tutorial 5: Molecule Generation Overview
  • Tutorial 6: Structure-Guided Generation
  • Tutorial 7: Property-Directed Generation
  • Tutorial 8: Prediction and Evaluation Modes
  • Tutorial 9: Analyze Module - 3D Molecular Structure Analysis

API Reference

  • API Reference
    • MolecularDiffusion
      • MolecularDiffusion.callbacks
        • MolecularDiffusion.callbacks.train_helper
      • MolecularDiffusion.cli
        • MolecularDiffusion.cli.analyze
        • MolecularDiffusion.cli.data
        • MolecularDiffusion.cli.eval_predict
        • MolecularDiffusion.cli.generate
        • MolecularDiffusion.cli.main
        • MolecularDiffusion.cli.predict
        • MolecularDiffusion.cli.train
      • MolecularDiffusion.core
        • MolecularDiffusion.core.core
        • MolecularDiffusion.core.engine
        • MolecularDiffusion.core.logger
        • MolecularDiffusion.core.meter
      • MolecularDiffusion.data
        • MolecularDiffusion.data.component
          • MolecularDiffusion.data.component.dataset
          • MolecularDiffusion.data.component.feature
          • MolecularDiffusion.data.component.pharmacophore
          • MolecularDiffusion.data.component.pointcloud
        • MolecularDiffusion.data.dataloader
        • MolecularDiffusion.data.dataset
        • MolecularDiffusion.data.valency_tables
      • MolecularDiffusion.molcraftdiff
      • MolecularDiffusion.runmodes
        • MolecularDiffusion.runmodes.analyze
          • MolecularDiffusion.runmodes.analyze.compare_to_optimized
          • MolecularDiffusion.runmodes.analyze.compute_metrics
          • MolecularDiffusion.runmodes.analyze.xtb_electronic
          • MolecularDiffusion.runmodes.analyze.xtb_optimization
          • MolecularDiffusion.runmodes.analyze.xyz2mol
        • MolecularDiffusion.runmodes.generate
          • MolecularDiffusion.runmodes.generate.tasks_generate
        • MolecularDiffusion.runmodes.train
          • MolecularDiffusion.runmodes.train.data
          • MolecularDiffusion.runmodes.train.eval
          • MolecularDiffusion.runmodes.train.logger
          • MolecularDiffusion.runmodes.train.tasks_egcl
          • MolecularDiffusion.runmodes.train.tasks_egt
          • MolecularDiffusion.runmodes.train.trainer
      • MolecularDiffusion.utils
        • MolecularDiffusion.utils.comm
        • MolecularDiffusion.utils.diffusion_utils
        • MolecularDiffusion.utils.file
        • MolecularDiffusion.utils.geom_analyzer
        • MolecularDiffusion.utils.geom_constant
        • MolecularDiffusion.utils.geom_constraint
        • MolecularDiffusion.utils.geom_metrics
        • MolecularDiffusion.utils.geom_stability
        • MolecularDiffusion.utils.geom_utils
        • MolecularDiffusion.utils.io
        • MolecularDiffusion.utils.molgraph_utils
        • MolecularDiffusion.utils.plot_function
        • MolecularDiffusion.utils.pretty
        • MolecularDiffusion.utils.sascore
        • MolecularDiffusion.utils.shepherd_score
          • MolecularDiffusion.utils.shepherd_score.alignment
            • MolecularDiffusion.utils.shepherd_score.alignment.utils
              • MolecularDiffusion.utils.shepherd_score.alignment.utils.pca
              • MolecularDiffusion.utils.shepherd_score.alignment.utils.pca_np
              • MolecularDiffusion.utils.shepherd_score.alignment.utils.se3
              • MolecularDiffusion.utils.shepherd_score.alignment.utils.se3_np
          • MolecularDiffusion.utils.shepherd_score.extract_profiles
          • MolecularDiffusion.utils.shepherd_score.generate_point_cloud
          • MolecularDiffusion.utils.shepherd_score.pharm_utils
            • MolecularDiffusion.utils.shepherd_score.pharm_utils.pharmacophore
            • MolecularDiffusion.utils.shepherd_score.pharm_utils.pharmvec
          • MolecularDiffusion.utils.shepherd_score.score
            • MolecularDiffusion.utils.shepherd_score.score.analytical_gradients
            • MolecularDiffusion.utils.shepherd_score.score.constants
            • MolecularDiffusion.utils.shepherd_score.score.electrostatic_scoring
            • MolecularDiffusion.utils.shepherd_score.score.electrostatic_scoring_np
            • MolecularDiffusion.utils.shepherd_score.score.gaussian_overlap
            • MolecularDiffusion.utils.shepherd_score.score.gaussian_overlap_np
            • MolecularDiffusion.utils.shepherd_score.score.pharmacophore_scoring
            • MolecularDiffusion.utils.shepherd_score.score.pharmacophore_scoring_np
        • MolecularDiffusion.utils.shepherd_utils
        • MolecularDiffusion.utils.smilify
        • MolecularDiffusion.utils.torch
        • MolecularDiffusion.utils.xyz2mol
          • MolecularDiffusion.utils.xyz2mol.converter
          • MolecularDiffusion.utils.xyz2mol.elements
Back to top
View this page
Edit this page

MolecularDiffusion.modules.models¶

Submodules¶

  • MolecularDiffusion.modules.models.egcl
  • MolecularDiffusion.modules.models.egt
  • MolecularDiffusion.modules.models.en_diffusion
  • MolecularDiffusion.modules.models.ldm
  • MolecularDiffusion.modules.models.noisemodel
  • MolecularDiffusion.modules.models.pharmacophore_dynamics
  • MolecularDiffusion.modules.models.shepherd_arch
  • MolecularDiffusion.modules.models.tabasco
Copyright © 2025, pregHosh
Made with Sphinx and @pradyunsg's Furo
On this page
  • MolecularDiffusion.modules.models
    • Submodules