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/md22_datamodule.py
nequip/data/datamodule/rmd17_datamodule.py
nequip/data/datamodule/sgdml_datamodule.py
nequip/data/datamodule/tm23_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/_test_data.py
nequip/data/dataset/_utils.py
nequip/data/dataset/lmdb_dataset.py
nequip/data/dataset/npz_dataset.py
nequip/data/transforms/__init__.py
nequip/data/transforms/neighborlist.py
nequip/data/transforms/stress_utils.py
nequip/data/transforms/type_mapper.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/saved_models/__init__.py
nequip/model/saved_models/_utils.py
nequip/model/saved_models/checkpoint.py
nequip/model/saved_models/package.py
nequip/nn/__init__.py
nequip/nn/_graph_mixin.py
nequip/nn/_tp_scatter_base.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/pair_potential.py
nequip/nn/rescale.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/scripts/__init__.py
nequip/scripts/_compile_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/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/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/modules.py
nequip/utils/resolvers.py
nequip/utils/test.py
nequip/utils/versions.py
nequip/utils/unittests/__init__.py
nequip/utils/unittests/conftest.py
nequip/utils/unittests/model_tests.py