LICENSE.md
README.rst
pyproject.toml
TACE.egg-info/PKG-INFO
TACE.egg-info/SOURCES.txt
TACE.egg-info/dependency_links.txt
TACE.egg-info/entry_points.txt
TACE.egg-info/requires.txt
TACE.egg-info/top_level.txt
tace/__init__.py
tace/dataset/__init__.py
tace/dataset/dataloader.py
tace/dataset/datamodule.py
tace/dataset/element.py
tace/dataset/graph.py
tace/dataset/neighbour_list.py
tace/dataset/quantity.py
tace/dataset/read.py
tace/dataset/sampler.py
tace/dataset/split.py
tace/dataset/statistics.py
tace/dataset/utils.py
tace/foundations/__init__.py
tace/foundations/download_link.py
tace/foundations/u_shift.py
tace/interface/__init__.py
tace/interface/ase/__init__.py
tace/interface/ase/calculator.py
tace/interface/lammps/__init__.py
tace/interface/lammps/mliap.py
tace/interface/openmm/__init__.py
tace/interface/torchsim/__init__.py
tace/interface/torchsim/torchsim.py
tace/interface/uspex26/__init__.py
tace/interface/uspex26/relax.py
tace/lightning/__init__.py
tace/lightning/lit_model.py
tace/lightning/lora.py
tace/lightning/los_skip.py
tace/lightning/torch_model.py
tace/lightning/trainer.py
tace/lightning/u_shift.py
tace/models/__init__.py
tace/models/adapter.py
tace/models/angular.py
tace/models/blocks.py
tace/models/ictd.py
tace/models/lammps.py
tace/models/layout.py
tace/models/legacy.py
tace/models/linear.py
tace/models/mlp.py
tace/models/normalizer.py
tace/models/precision.py
tace/models/radial.py
tace/models/s2.py
tace/models/softmax.py
tace/models/utils.py
tace/models/zemin.py
tace/models/_cart/__init__.py
tace/models/_cart/base.py
tace/models/_cart/basis_change.py
tace/models/_cart/cartesian.py
tace/models/_cart/default.py
tace/models/_cart/dropout.py
tace/models/_cart/edge.py
tace/models/_cart/fused.py
tace/models/_cart/inter.py
tace/models/_cart/layer_norm.py
tace/models/_cart/node.py
tace/models/_cart/nonlinear.py
tace/models/_cart/paths.py
tace/models/_cart/prod.py
tace/models/_cart/readout.py
tace/models/_cart/representation.py
tace/models/_cart/residual.py
tace/models/_cart/tace.py
tace/models/_cart/ue.py
tace/models/_cue/__init__.py
tace/models/_cue/_tp_scatter.py
tace/models/_cue/_tp_uuu.py
tace/models/_cue/paths.py
tace/models/_e3nn/__init__.py
tace/models/_e3nn/asymmetric_contraction.py
tace/models/_e3nn/base.py
tace/models/_e3nn/basis_change.py
tace/models/_e3nn/default.py
tace/models/_e3nn/dropout.py
tace/models/_e3nn/edge.py
tace/models/_e3nn/fused.py
tace/models/_e3nn/inter.py
tace/models/_e3nn/layer_norm.py
tace/models/_e3nn/node.py
tace/models/_e3nn/nonlinear.py
tace/models/_e3nn/paths.py
tace/models/_e3nn/prod.py
tace/models/_e3nn/readout.py
tace/models/_e3nn/representation.py
tace/models/_e3nn/residual.py
tace/models/_e3nn/scatter_norm.py
tace/models/_e3nn/symmetric_contraction.py
tace/models/_e3nn/tace.py
tace/models/_e3nn/ue.py
tace/models/_eqt/__init__.py
tace/models/_eqt/_tp_uuu.py
tace/models/_eqt/equitorch/__init__.py
tace/models/_eqt/equitorch/irreps/__init__.py
tace/models/_eqt/equitorch/nn/__init__.py
tace/models/_eqt/equitorch/nn/sparse_product.py
tace/models/_eqt/equitorch/nn/tensor_products.py
tace/models/_eqt/equitorch/structs/__init__.py
tace/models/_oeq/__init__.py
tace/models/_oeq/_tp_scatter.py
tace/models/compile/__init__.py
tace/models/compile/aot.py
tace/models/compile/compile.py
tace/models/compile/tace.py
tace/models/compile/wrapper.py
tace/models/so2/__init__.py
tace/models/so2/blocks.py
tace/models/so2/rotation_matrix.py
tace/models/so2/utils.py
tace/models/so2/wigner.py
tace/models/triton_ops/__init__.py
tace/models/triton_ops/uu_so2_scatter.py
tace/scripts/__init__.py
tace/scripts/average.py
tace/scripts/clean.py
tace/scripts/convert.py
tace/scripts/eval.py
tace/scripts/export_eval.py
tace/scripts/export_lammps.py
tace/scripts/export_train.py
tace/scripts/finetune.py
tace/scripts/split.py
tace/scripts/train.py
tace/scripts/update.py
tace/utils/__init__.py
tace/utils/_global.py
tace/utils/callbacks.py
tace/utils/cfg.py
tace/utils/ema.py
tace/utils/env.py
tace/utils/hydra_resolver.py
tace/utils/logger.py
tace/utils/mask_metrics.py
tace/utils/metrics.py
tace/utils/metrics_bake.py
tace/utils/spectra.py
tace/utils/strategy.py
tace/utils/torch_scatter.py
tace/utils/units.py
tace/utils/utils.py
tace/utils/loss/__init__.py
tace/utils/loss/common.py
tace/utils/loss/dens.py
tace/utils/loss/huber_fn.py
tace/utils/loss/l2mae_fn.py
tace/utils/loss/mae_fn.py
tace/utils/loss/mse_fn.py
tace/utils/loss/normal.py
tace/utils/loss/registry.py
tace/utils/loss/special_fn.py
tace/utils/loss/uncertainty.py
tace/utils/lr_scheduler/__init__.py
tace/utils/lr_scheduler/warmup.py
tace/utils/lr_scheduler/wsd.py
tace/utils/optimizer/__init__.py
tace/utils/optimizer/hybrid_muon.py
tace/utils/optimizer/soap.py
test/test_aoti_single_system.py
test/test_neighbour_list.py
test/test_solid_harmonics.py
test/test_wigner6j.py