Skip to content

Commit f9fc2e9

Browse files
authored
Model Refactor (#66)
* initial implementation of pipeline wrapper and model updates * merge model card and config, update docs * update examples readme * review comments and model cleanup * expose a better neighbor list mechanism and fix some aimnet + pipeline issues, add example * make some examples run and update aimnet to use neighbors utilities * update aimnet, other review comments * more review responses and improve aimnet coverage * preempt merge conflict in 01 example * bug in pipeline and aimnet2 * more merge conflict resolutions * fix model_config mutation in pipeline * fix missing skin argument * remove errant neighbor list file * add packages and fix mace test bug * fix skin dropping
1 parent bb2e1d5 commit f9fc2e9

71 files changed

Lines changed: 8583 additions & 4700 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

docs/conf.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,6 @@
7474
"sphinx_togglebutton",
7575
"sphinx.ext.graphviz",
7676
"sphinx_gallery.gen_gallery",
77-
"model_card_ext",
7877
]
7978

8079
source_suffix = [".rst", ".md"]
@@ -86,7 +85,6 @@
8685
exclude_patterns = [
8786
"_build",
8887
"sphinxext.py",
89-
"model_card_ext.py",
9088
"Thumbs.db",
9189
".DS_Store",
9290
]

docs/index.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ batching and size-aware sampling handle load balancing automatically.
4646
## Highlights
4747

4848
- **Bring your own model** --- wrap MACE, AIMNet2, or any PyTorch MLIP in a
49-
few lines with a standardized `ModelCard` interface.
49+
few lines with a standardized `ModelConfig` interface.
5050
- **Compose, don't configure** --- fuse stages on one GPU with `+`, distribute
5151
across GPUs with `|`, and inject behavior at nine hook points per step.
5252
- **GPU-native data** --- `AtomicData` and `Batch` are Pydantic-validated,

docs/model_card_ext.py

Lines changed: 0 additions & 319 deletions
This file was deleted.

0 commit comments

Comments
 (0)