GitHub repository: Aiqing-Zhu/onsagernet-jax
Primary language: Python
Description: jax implementation of OnsagerNet