Chemeleon2 是一个用于晶体结构生成的强化学习框架，它在潜在扩散模型中使用群体相对策略优化

该项目实现了一个三阶段的晶体结构生成流程：

1. VAE 模块：将晶体结构编码到潜在空间表示中
2. LDM 模块：使用扩散 Transformer 在潜在空间中采样晶体结构
3. RL 模块：通过自定义奖励函数对 LDM 进行微调，以优化特定的材料性质

该框架的核心优势是允许用户设计自定义奖励函数，通过简单的 Python 接口引导生成过程朝向期望的材料性质（如带隙、密度、稳定性等）。

项目还提供了多个内置的奖励组件，包括：

1. CreativityReward：奖励独特和新颖的结构
2. EnergyReward：惩罚高于凸包的高能量
3. StructureDiversityReward：鼓励多样化的晶体几何结构
4. CompositionDiversityReward：鼓励多样化的化学组成
5. PredictorReward：使用训练好的机器学习模型作为奖励函数

该框架专为定向材料发现而设计，用户可以通过定义自定义奖励函数将晶体生成引导到具有特定材料性质的方向。