Microjax: JAX en dos clases y seis funciones

Inspirado en Micrograd de Andrej Karpathy, Microjax es una biblioteca que replica la funcionalidad de JAX usando solo dos clases y seis funciones. A diferencia del popular PyTorch, Microjax adopta el estilo de programación más funcional de JAX. Este tutorial toma mucho de la excelente presentación de Matthew J Johnson de 2017 sobre autograd, el predecesor de JAX, simplificándolo y empaquetándolo como un notebook.