mattergen
MatterGen 官方實作：一個針對無機材料設計的生成模型，可在整個週期表中操作，並能微調以引導生成結果符合多樣化的材料屬性限制。
mattergen 是什麼？
MatterGen 的官方實作。此生成模型專為無機材料設計而開發，能於整個週期表中運作，並可經由微調來引導生成結果符合多樣化的材料屬性限制。
mattergen 概覽
這是什麼
MatterGen 是一個用於無機材料設計的生成模型，可跨越整個元素週期表進行生成，並可透過微調來引導生成朝向多種性質約束。此儲存庫為官方實作。
主要功能
- 跨越元素週期表的無機材料生成模型
- 支援微調以根據性質條件進行生成，例如化學系統、空間群、能隙、磁密度、體積模量與高於 hull 的能量
- 提供預訓練的無條件模型與微調模型檢查點
- 包含使用 MatterSim 機器學習力場進行結構弛豫與穩定性評估的腳本
- 資料集與模型檢查點透過 Git Large File Storage (LFS) 提供，也可在 Hugging Face 取得
使用情境
- 無條件生成新型無機晶體結構
- 根據性質條件生成（例如目標磁密度、能隙、體積模量）
- 多性質條件生成（例如化學系統與高於 hull 的能量）
- 評估生成結構的新穎性、獨特性、穩定性及其他指標
快速開始
透過 uv 安裝依賴（假設使用 Linux 與 CUDA GPU）：bashpip install uvuv venv .venv --python 3.10source .venv/bin/activateuv pip install -e .或者從 PyPI 安裝，搭配預先建置的 PyTorch Geometric CUDA wheels：bashuv pip install mattergen --find-links https://data.pyg.org/whl/torch-2.2.0+cu121.html需要 Git LFS 來取得檢查點與資料集。若未安裝：bashsudo apt install git-lfsgit lfs install然後拉取檢查點：bashgit lfs pull -I checkpoints/ --exclude=''無條件生成範例：bashexport MODEL_NAME=mattergen_baseexport RESULTS_PATH=results/mattergen-generate $RESULTS_PATH --pretrained-name=$MODEL_NAME --batch_size=16 --num_batches 1性質條件生成（例如磁密度）：bashexport MODEL_NAME=dft_mag_densityexport RESULTS_PATH='results/$MODEL_NAME/'mattergen-generate $RESULTS_PATH --pretrained-name=$MODEL_NAME --batch_size=16 --properties_to_condition_on='{'dft_mag_density': 0.15}' --diffusion_guidance_factor=2.0
授權與價格
授權：MIT。價格類型：開放原始碼。
內容整理自該專案公開的 README 與儲存庫資訊。
