Add aimnet2-pd model weights and card

README.md

@@ -0,0 +1,106 @@
+---
+library_name: aimnet
+license: mit
+tags:
+  - chemistry
+  - computational-chemistry
+  - molecular-dynamics
+  - interatomic-potential
+  - neural-network-potential
+  - pytorch
+  - aimnet2
+pipeline_tag: other
+---
+
+# AIMNet2 Pd (Palladium)
+
+AIMNet2 is a neural network interatomic potential for fast and accurate molecular simulations. This is an extended model supporting **palladium-containing systems** for organometallic catalysis and coordination chemistry.
+
+## Highlights
+
+- **Fast:** Orders of magnitude faster than DFT, with optional `torch.compile` for ~5x additional speedup on GPU
+- **Accurate:** Near-DFT accuracy for energies, forces, and charges
+- **Broad coverage:** Supports 14 elements: H, B, C, N, O, F, Si, P, S, Cl, Se, Br, Pd, I
+- **Ensemble:** 4 ensemble members for uncertainty estimation
+- **Features:** Energy, forces, charges, Hessian, stress tensor, periodic boundary conditions
+
+## Installation
+
+```bash
+pip install "aimnet[hf]"
+```
+
+## Quick Start
+
+```python
+from aimnet.calculators import AIMNet2Calculator
+
+# Load from Hugging Face (downloads and caches automatically)
+calc = AIMNet2Calculator("isayevlab/aimnet2-pd")
+
+# Single-point calculation
+results = calc(
+    {"coord": coords, "numbers": atomic_numbers, "charge": 0.0},
+    forces=True,
+)
+print(results["energy"])   # Energy in eV
+print(results["forces"])   # Forces in eV/A
+print(results["charges"])  # Partial charges in e
+```
+
+### With ASE
+
+```python
+from aimnet.calculators.aimnet2ase import AIMNet2ASE
+from ase.build import molecule
+
+atoms = molecule("H2O")
+atoms.calc = AIMNet2ASE("isayevlab/aimnet2-pd")
+
+energy = atoms.get_potential_energy()
+forces = atoms.get_forces()
+```
+
+## Ensemble Members
+
+This repo contains 4 ensemble members (`ensemble_0.safetensors` through `ensemble_3.safetensors`). The default loads member 0. For uncertainty estimation, load all 4 and average predictions.
+
+| File | Description |
+|------|-------------|
+| `ensemble_0.safetensors` | Ensemble member 0 (default) |
+| `ensemble_1.safetensors` | Ensemble member 1 |
+| `ensemble_2.safetensors` | Ensemble member 2 |
+| `ensemble_3.safetensors` | Ensemble member 3 |
+| `config.json` | Shared model configuration and metadata |
+
+## Model Details
+
+- **Architecture:** AIMNet2 (Atomic Environment Vectors + message-passing MLPs)
+- **Cutoff radius:** 5.0 A
+- **DFT functional:** wB97M-D3
+- **Dispersion:** D3BJ (externalized, handled by calculator)
+- **Coulomb:** No embedded short-range Coulomb (coulomb_mode=none)
+- **Format:** safetensors (converted from PyTorch state dict)
+
+## Limitations
+
+- **Periodic systems** are supported through the Python API / ASE interface, not through the Gradio demo
+- Element coverage is limited to the 14 elements listed above; unsupported elements will raise an error
+- This model does not include arsenic (As) — use `aimnet2-wb97m-d3` for As-containing systems
+
+## Citation
+
+```bibtex
+@article{anstine2025aimnet2,
+  title={AIMNet2: A Neural Network Potential to Meet your Neutral, Charged, Organic, and Elemental-Organic Needs},
+  author={Anstine, Dylan and Zubatyuk, Roman and Isayev, Olexandr},
+  journal={Chemical Science},
+  year={2025},
+  publisher={Royal Society of Chemistry},
+  doi={10.1039/D4SC08572H}
+}
+```
+
+## License
+
+MIT License

config.json

@@ -0,0 +1,38 @@
+{
+  "config_schema_version": 1,
+  "family_name": "aimnet2-pd",
+  "ensemble_size": 4,
+  "member_names": [
+    "aimnet2-pd_0",
+    "aimnet2-pd_1",
+    "aimnet2-pd_2",
+    "aimnet2-pd_3"
+  ],
+  "cutoff": 5.0,
+  "needs_coulomb": false,
+  "needs_dispersion": true,
+  "coulomb_mode": "none",
+  "implemented_species": [
+    1,
+    5,
+    6,
+    7,
+    8,
+    9,
+    14,
+    15,
+    16,
+    17,
+    34,
+    35,
+    46,
+    53
+  ],
+  "library_name": "aimnet",
+  "tags": [
+    "chemistry",
+    "molecular-dynamics",
+    "force-field",
+    "aimnet2"
+  ]
+}

ensemble_0.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4582291ab07a4afcb81fe9920b649304ba81eced98d55a45973a2e96da269162
+size 8825564

ensemble_1.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c701e06468f8dc131e2d5fc26d1211732358f6c71e3e1218821180ca2d2da1e4
+size 8825564

ensemble_2.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5d47a783a923e4e423b6111c462f64e79a2b6e316ab5ff62096f5b73b2529f1
+size 8825564

ensemble_3.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:37097aac0373021d77a7baeca5503b90fc26d3427b3db9c56a418d60f0268d84
+size 8825564