nequip.data

• Data Fields
  • _GRAPH_FIELDS
  • _NODE_FIELDS
  • _EDGE_FIELDS
  • _LONG_FIELDS
  • _CARTESIAN_TENSOR_FIELDS
  • register_fields()
  • deregister_fields()
• nequip.data.datamodule
  • NequIPDataModule
  • ASEDataModule
  • sGDML_CCSD_DataModule
  • NequIP3BPADataModule
• nequip.data.dataset
  • AtomicDataset
  • NequIPLMDBDataset
  • ASEDataset
  • HDF5Dataset
  • EMTTestDataset
  • SubsetByRandomSlice
  • RandomSplitAndIndexDataset()
• nequip.data.transforms
  • ChemicalSpeciesToAtomTypeMapper
  • NeighborListTransform
  • VirialToStressTransform
• Data Modifiers
  • PerAtomModifier
  • EdgeLengths
  • NumNeighbors
• Dataset Statistics
  • DataStatisticsManager
  • Mean
  • MeanAbsolute
  • RootMeanSquare
  • StandardDeviation
  • Min
  • Max
  • Count