Jumpy

Jumpy是JAX或numpy的通用后端。

• Jumpy函数如果给定JAX输入，则返回JAX输出。
• 跳转函数在JAX编译时返回JAX输出。
• 否则，跳转函数返回NumPy输出。

Jumpy允许您编写与框架无关的代码，可以通过直接运行NumPy进行调试，但JAX编译时性能与JAX相当。

安装Jumpy

要从pypi安装Jumpy

python3 -m venv env
source env/bin/activate
pip install --upgrade pip
pip install brax-jumpy

或者，要从源安装Jumpy，克隆此存储库，cd到它，然后

python3 -m venv env
source env/bin/activate
pip install --upgrade pip
pip install -e .