程序初版完成，目前收敛后结果不太对

2025-10-23 10:29:27 +08:00
parent 6694450c70
commit 64ca5e257a
9 changed files with 913 additions and 94 deletions
@@ -1,3 +1,6 @@
+# JAX 入门示例代码
+# 演示 JAX 的基本用法，包括 jnp 数组、jit 加速
+
 import numpy as np
 import jax
 import jax.numpy as jnp