AUTHORS.md
LICENSE
README.md
pyproject.toml
nequip/__init__.py
nequip/_version.py
nequip.egg-info/PKG-INFO
nequip.egg-info/SOURCES.txt
nequip.egg-info/dependency_links.txt
nequip.egg-info/entry_points.txt
nequip.egg-info/requires.txt
nequip.egg-info/top_level.txt
nequip/ase/__init__.py
nequip/ase/nequip_calculator.py
nequip/ase/nosehoover.py
nequip/data/AtomicDataDict.py
nequip/data/__init__.py
nequip/data/_key_registry.py
nequip/data/_keys.py
nequip/data/_nl.py
nequip/data/_sampler.py
nequip/data/ase.py
nequip/data/dict.py
nequip/data/misc.py
nequip/data/modifier.py
nequip/data/stats.py
nequip/data/stats_manager.py
nequip/data/datamodule/_3bpa_datamodule.py
nequip/data/datamodule/__init__.py
nequip/data/datamodule/_ase_datamodule.py
nequip/data/datamodule/_base_datamodule.py
nequip/data/datamodule/coll_datamodule.py
nequip/data/datamodule/md22_datamodule.py
nequip/data/datamodule/rmd17_datamodule.py
nequip/data/datamodule/samd23_datamodule.py
nequip/data/datamodule/sgdml_datamodule.py
nequip/data/datamodule/tm23_datamodule.py
nequip/data/datamodule/water_datamodule.py
nequip/data/dataset/__init__.py
nequip/data/dataset/ase_dataset.py
nequip/data/dataset/base_datasets.py
nequip/data/dataset/hdf5_dataset.py
nequip/data/dataset/lmdb_dataset.py
nequip/data/dataset/npz_dataset.py
nequip/data/dataset/test_data.py
nequip/data/dataset/utils.py
nequip/data/dataset_stats/MPTrj.yaml
nequip/data/dataset_stats/MatPES.yaml
nequip/data/dataset_stats/OMat.yaml
nequip/data/transforms/__init__.py
nequip/data/transforms/cell_utils.py
nequip/data/transforms/neighborlist.py
nequip/data/transforms/stress_utils.py
nequip/data/transforms/type_mapper.py
nequip/integrations/__init__.py
nequip/integrations/torchsim.py
nequip/integrations/lammps_mliap/create_lmp_mliap_file.py
nequip/integrations/lammps_mliap/lmp_mliap_wrapper.py
nequip/model/__init__.py
nequip/model/modify_utils.py
nequip/model/nequip_models.py
nequip/model/pair_potential.py
nequip/model/utils.py
nequip/model/inference_models/__init__.py
nequip/model/inference_models/aotinductor.py
nequip/model/inference_models/compiled.py
nequip/model/inference_models/torchscript.py
nequip/model/saved_models/__init__.py
nequip/model/saved_models/_utils.py
nequip/model/saved_models/checkpoint.py
nequip/model/saved_models/load_utils.py
nequip/model/saved_models/package.py
nequip/nn/__init__.py
nequip/nn/_ghost_exchange_base.py
nequip/nn/_ghost_exchange_lmp_mliap.py
nequip/nn/_graph_mixin.py
nequip/nn/_tp_scatter_base.py
nequip/nn/_tp_scatter_cueq.py
nequip/nn/_tp_scatter_oeq.py
nequip/nn/atomwise.py
nequip/nn/compile.py
nequip/nn/convnetlayer.py
nequip/nn/grad_output.py
nequip/nn/graph_model.py
nequip/nn/interaction_block.py
nequip/nn/misc.py
nequip/nn/mlp.py
nequip/nn/model_modifier_utils.py
nequip/nn/nonlinearities.py
nequip/nn/norm.py
nequip/nn/pair_potential.py
nequip/nn/utils.py
nequip/nn/embedding/__init__.py
nequip/nn/embedding/_edge.py
nequip/nn/embedding/cutoffs.py
nequip/nn/embedding/node.py
nequip/nn/embedding/utils.py
nequip/scripts/__init__.py
nequip/scripts/_compile_utils.py
nequip/scripts/_package_utils.py
nequip/scripts/_workflow_utils.py
nequip/scripts/compile.py
nequip/scripts/package.py
nequip/scripts/train.py
nequip/train/__init__.py
nequip/train/config.py
nequip/train/ema.py
nequip/train/lightning.py
nequip/train/metrics.py
nequip/train/metrics_manager.py
nequip/train/schedulefree.py
nequip/train/simple_ddp.py
nequip/train/callbacks/__init__.py
nequip/train/callbacks/loss_coeff_monitor.py
nequip/train/callbacks/loss_coeff_scheduler.py
nequip/train/callbacks/softadapt.py
nequip/train/callbacks/tf32_scheduler.py
nequip/train/callbacks/training_stats.py
nequip/train/callbacks/wandb_watch.py
nequip/train/callbacks/write_xyz.py
nequip/utils/__init__.py
nequip/utils/aot.py
nequip/utils/compile.py
nequip/utils/dtype.py
nequip/utils/file_utils.py
nequip/utils/fx.py
nequip/utils/global_dtype.py
nequip/utils/global_state.py
nequip/utils/logger.py
nequip/utils/model_cache.py
nequip/utils/model_repository.py
nequip/utils/modules.py
nequip/utils/resolvers.py
nequip/utils/test.py
nequip/utils/unittests/__init__.py
nequip/utils/unittests/conftest.py
nequip/utils/unittests/minimal_aspirin.yaml
nequip/utils/unittests/minimal_toy_emt.yaml
nequip/utils/unittests/model_tests.py
nequip/utils/unittests/model_tests_basic.py
nequip/utils/unittests/model_tests_compilation.py
nequip/utils/unittests/model_tests_lammps.py
nequip/utils/unittests/model_tests_torchsim.py
nequip/utils/unittests/utils.py
nequip/utils/versions/__init__.py
nequip/utils/versions/package_versions.py
nequip/utils/versions/torch_versions.py
nequip/utils/versions/version_utils.py