JAX: Composable transformations of Python+NumPy programs, James Bradbury, Roy Frostig, Peter Hawkins, Matthew James Johnson, Chris Leary, Skye Lew, George H. Nash, Nicolas Nieto, Sam Oden, Mark Omernick, Jonathan Renjie So, Matt Von-Rose and Sasha Rush, 2018 - 介绍JAX库的原始参考文献,概述了其自动微分和其他程序转换的设计原则。