使用JAX连接数值相对论与自动微分
原文英文，约4500词，阅读约需17分钟。发表于：。Introduction Albert Einstein's general theory of relativity is the most successful picture of how gravity works that we have as of today. It underlies our current understanding of the physics of...
自动微分是一种高效计算函数导数的方法，结合了符号和数值微分的优点。它通过构建有向无环图并使用链式法则计算梯度，广泛应用于深度学习库如TensorFlow和PyTorch。JAX是支持自动微分的高性能库，适用于机器学习和科学计算。本文探讨了自动微分在数值相对论中的应用，展示了如何用JAX计算度量张量的导数。
