我正在寻找用 Python 编写的库，这将使我能够求解耦合的非线性方程，如下所示：

我需要能够让我将此求解器与其他模型耦合的库。例如，我想施加不同的边界条件并及时更改它们（在任何时间步）以及在每个时间步访问解向量并有可能更改它（例如，实现非本地传输）。

其他“愿望”：

• 快速地
• 有并行求解的可能。
• 如果可能的话，易于使用。

先感谢您！