diffjeom

由jax驱动的微分几何。给定一个度量函数，此包允许您计算

• 第二类克里斯托费尔符号 (get_christoffel2);
• 黎曼曲率张量 (get_riemann);
• 里奇张量 (get_ricci_tensor);
• 里奇标量/标量曲率 (get_ricci_scalar).

这些可以使用jax的常规转换，如jit和vmap。

提供函数check_christoffel2_sym、check_riemann_sym、check_diff_bianchi和check_ricci_tensor_sym来检查这些对象的恒等式。

入门

使用以下方式安装

pip install diffjeom

如果您还没有安装jax和jaxlib，这将安装它们。测试需要numpy，可以使用pytest运行。查看它们以获取一些使用示例。