atomgen namespace#
Subpackages#
- atomgen.data package
- atomgen.models package
- Submodules
- atomgen.models.configuration_atomformer module
- atomgen.models.modeling_atomformer module
AtomFormerForSystemClassification
AtomformerEncoder
AtomformerForCoordinateAM
AtomformerForMaskedAM
AtomformerModel
AtomformerPreTrainedModel
GaussianLayer
InitialStructure2RelaxedEnergy
InitialStructure2RelaxedStructure
InitialStructure2RelaxedStructureAndEnergy
ParallelBlock
Structure2Energy
Structure2EnergyAndForces
Structure2Forces
Structure2TotalEnergyAndForces
- atomgen.models.schnet module
- atomgen.models.tokengt module
- Submodules