A dead simple Python package for creating custom JAX pytree objects - View it on GitHub