JAX的设计优劣评述

JAX 是一个设计良好的低级机器学习框架应有的样子。好的设计能让你用更少的努力实现更高的性能。糟糕的设计恰恰相反。