机器学习库JAX

目前我们用于机器学习的数值计算库主要还是numpy，但是numpy的函数太过于庞杂，需要大量的时间来熟悉；而这款JAX是基于numpy的高性能数值计算库，目前已经在机器学习领域崭露…