spydr is a Python port of the machine learning research library spidr. While spidr has its own tensor API that uses XLA directly for its backend, spydr uses JAX which provides its own NumPy array API powered by XLA (and Autograd). Otherwise, spidr and spydr are largely identical.
Clone this repository, then run
$ pip install .in the repository root.