非弹性本构人工神经网络的广义对偶势：有限应变下的 JAX 实现