This is a toy/learning project to explore autodiffing and JIT via XLA, inspired by Jax and its Autodidax.