Microjax：仅用两类六函数实现JAX功能

受Andrej Karpathy的Micrograd启发，作者创建了Microjax，一个仅用两类六个函数就实现了类似JAX功能的库。不同于流行的PyTorch，Microjax采用更具函数式编程风格的JAX。该教程大量借鉴了Matthew J Johnson 2017年关于JAX前身autograd的精彩演示，并将其简化并打包成一个notebook。