paddle 无法使用jit比jax训练慢

### 需求描述 Feature Description

**任务目标**：复现[CausalPINN](https://github.com/PredictiveIntelligenceLab/CausalPINNs/blob/main/NS/NS.py)代码

**需求场景** 每次迭代训练速度paddle比 _jit()_ 下的jax慢三倍（0.1s vs 0.3s）

**功能描述** 在paddle上实现 _jit()_ 函数

