AI交流(进群备注:eqnn-jax)

基于Jax实现的等变图神经网络库,为复杂结构数据建模提供强大的工具。

eqnn-jax的特点:

  • 1. 支持多种等变模型(EGNN、SEGNN、NequIP等),满足多样化需求
  • 2. 提供详细的注释和示例代码,易于上手
  • 3. 包含宇宙学基准测试数据集,拓展应用场景

eqnn-jax的功能:

  • 1. 用于复杂结构数据的建模
  • 2. 在科研中进行等变图神经网络的实验和开发
  • 3. 利用宇宙学数据集进行深度学习研究

相关导航

暂无评论

暂无评论...