Microjax : JAX en deux classes et six fonctions

Inspiré par Micrograd d'Andrej Karpathy, Microjax est une bibliothèque qui reproduit les fonctionnalités de JAX en utilisant seulement deux classes et six fonctions. Contrairement au populaire PyTorch, Microjax adopte le style de programmation plus fonctionnel de JAX. Ce tutoriel s'inspire largement de l'excellente présentation de Matthew J Johnson de 2017 sur autograd, le prédécesseur de JAX, en le simplifiant et en le proposant sous forme de notebook.