LICENSE
NOTICE
README.md
pyproject.toml
src/flashcart/__init__.py
src/flashcart/py.typed
src/flashcart.egg-info/PKG-INFO
src/flashcart.egg-info/SOURCES.txt
src/flashcart.egg-info/dependency_links.txt
src/flashcart.egg-info/entry_points.txt
src/flashcart.egg-info/requires.txt
src/flashcart.egg-info/top_level.txt
src/flashcart/calculators/__init__.py
src/flashcart/calculators/ase.py
src/flashcart/calculators/lammps_mliap.py
src/flashcart/cli/__init__.py
src/flashcart/cli/lammps_mliap.py
src/flashcart/cli/profile.py
src/flashcart/cli/test.py
src/flashcart/cli/train.py
src/flashcart/configs/default.yaml
src/flashcart/configs/profile.yaml
src/flashcart/data/__init__.py
src/flashcart/data/data.py
src/flashcart/data/dataset.py
src/flashcart/data/graph.py
src/flashcart/data/neighbors.py
src/flashcart/data/padding.py
src/flashcart/data/samplers.py
src/flashcart/data/statistics.py
src/flashcart/data/utils.py
src/flashcart/model/__init__.py
src/flashcart/model/atomistic.py
src/flashcart/model/flashcart.py
src/flashcart/nn/__init__.py
src/flashcart/nn/layers.py
src/flashcart/nn/radial.py
src/flashcart/o3/__init__.py
src/flashcart/o3/_codegen_cache.py
src/flashcart/o3/_codegen_common.py
src/flashcart/o3/_codegen_irreps.py
src/flashcart/o3/_codegen_linear.py
src/flashcart/o3/_codegen_tensor_product.py
src/flashcart/o3/_irreps.py
src/flashcart/o3/_linear.py
src/flashcart/o3/_tensor_product.py
src/flashcart/o3/_triton_launch.py
src/flashcart/o3/irreps.py
src/flashcart/o3/kernel_config.py
src/flashcart/o3/linear.py
src/flashcart/o3/tensor_product.py
src/flashcart/o3/utils.py
src/flashcart/training/__init__.py
src/flashcart/training/loss.py
src/flashcart/training/muon.py
src/flashcart/training/muon_norm.py
src/flashcart/training/optimizers.py
src/flashcart/training/schedulers.py
src/flashcart/training/tasks.py
src/flashcart/utils/__init__.py
src/flashcart/utils/compile.py
src/flashcart/utils/config.py
src/flashcart/utils/distributed.py
src/flashcart/utils/env.py
src/flashcart/utils/geometry.py
src/flashcart/utils/lammps.py
src/flashcart/utils/logging.py
src/flashcart/utils/parameter_groups.py
src/flashcart/utils/scatter.py
src/flashcart/utils/torch_geometric/__init__.py
src/flashcart/utils/torch_geometric/batch.py
src/flashcart/utils/torch_geometric/data.py
src/flashcart/utils/torch_geometric/dataloader.py
src/flashcart/utils/torch_geometric/dataset.py