JAX和可组合的程序转换
https://github.com/google/jax 的“关于”部分如下所述。 Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more “可组合变换”是什么意思?在 NeurIPS2020: JAX 生态系统 Meetup 的视频中,DeepMind 的一位工程师进行了解释。 举个例子,考虑以下函数。 def fn(x, y): return